From 2863a325585d449e3e3c8c0d6b9dcbfa58af650d Mon Sep 17 00:00:00 2001
From: Filippo Luca Ferretti <102977828+flferretti@users.noreply.github.com>
Date: Mon, 12 Feb 2024 12:18:39 +0100
Subject: [PATCH] Update src/jaxsim/simulation/integrators.py

Co-authored-by: Diego Ferigo <diego.ferigo@iit.it>
---
 src/jaxsim/simulation/integrators.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/jaxsim/simulation/integrators.py b/src/jaxsim/simulation/integrators.py
index 56928aa05..680532a39 100644
--- a/src/jaxsim/simulation/integrators.py
+++ b/src/jaxsim/simulation/integrators.py
@@ -38,7 +38,7 @@ class IntegratorType(enum.IntEnum):
 
 def integrator_fixed_single_step(
     dx_dt: StateDerivativeCallable,
-    x0: State,
+    x0: State | ODEState,
     t0: Time,
     tf: Time,
     integrator_type: IntegratorType,