Fixed another issue with the visualizer when the timestep is a np arr…

…ay with a single element
ami-iit · Apr 19, 2024 · 54540e6 · 54540e6
1 parent 356acd2
commit 54540e6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/hippopt/robot_planning/utilities/humanoid_state_visualizer.py b/src/hippopt/robot_planning/utilities/humanoid_state_visualizer.py
@@ -251,6 +251,8 @@ def _visualize_multiple_states(
             or _timestep_s.size == 1
         ):
             single_step = _timestep_s if _timestep_s is not None else 0.0
+            if isinstance(single_step, np.ndarray):
+                single_step = single_step.item()
             _timestep_s = [single_step] * len(states)
 
         if len(_timestep_s) != len(states):