Merge pull request #236 from LuziaKn/ft-add-general-visualization

fix error due to visualization id not matching number of visualizations
maxspahn · Oct 19, 2023 · 57b82e7 · 57b82e7
2 parents fdfe3d9 + 467461d
commit 57b82e7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/urdfenvs/urdf_common/urdf_env.py b/urdfenvs/urdf_common/urdf_env.py
@@ -340,8 +340,8 @@ def update_collision_links(self) -> None:
             )
 
     def update_visualizations(self, positions) -> None:
-        for visual_shape_id, info in self._visualizations.items():
-            position = positions[visual_shape_id-1]
+        for i, (visual_shape_id, info) in enumerate(self._visualizations.items()):
+            position = positions[i]
             rotation = [1, 0, 0, 0]
             p.resetBasePositionAndOrientation(
                 visual_shape_id, position, rotation