Update src/jaxsim/simulation/integrators.py

Co-authored-by: Diego Ferigo <[email protected]>
ami-iit · Feb 12, 2024 · a225c87 · a225c87
1 parent 410bf16
commit a225c87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/jaxsim/simulation/integrators.py b/src/jaxsim/simulation/integrators.py
@@ -65,7 +65,7 @@ def integrator_fixed_single_step(
     sub_step_dt = dt / num_sub_steps
 
     # Initialize the carry
-    Carry = Tuple[State, Time]
+    Carry = Tuple[State | ODEState, Time]
     carry_init: Carry = (x0, t0)
 
     def forward_euler_body_fun(carry: Carry, xs: None) -> Tuple[Carry, None]: