Update src/jaxsim/simulation/integrators.py

Co-authored-by: Diego Ferigo <[email protected]>

src/jaxsim/simulation/integrators.py

@@ -43,7 +43,7 @@ def integrator_fixed_single_step(
     tf: Time,
     integrator_type: IntegratorType,
     num_sub_steps: int = 1,
-) -> Tuple[State, Dict[str, Any]]:
+) -> Tuple[State | ODEState, Dict[str, Any]]:
     """
     Advance a state vector by integrating a sytem dynamics with a fixed-step integrator.