Update src/jaxsim/simulation/integrators.py

Co-authored-by: Diego Ferigo <[email protected]>
ami-iit · Feb 12, 2024 · 2863a32 · 2863a32
1 parent 3fbdcc1
commit 2863a32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,