[Bugfix] call set_base_pose in SapienPlanningWorld during init and up…

…date
haosulab · Aug 17, 2024 · 86376ca · 86376ca
1 parent 7986c16
commit 86376ca
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mplib/sapien_utils/conversion.py b/mplib/sapien_utils/conversion.py
@@ -99,6 +99,7 @@ def __init__(
                 joint_names=[j.name for j in articulation.active_joints],
                 verbose=False,
             )
+            articulated_model.set_base_pose(articulation.root_pose)  # type: ignore
             articulated_model.set_qpos(
                 articulation.qpos,  # type: ignore
                 full=True,
@@ -130,6 +131,7 @@ def update_from_simulation(self, *, update_attached_object: bool = True) -> None
         """
         for articulation in self._sim_scene.get_all_articulations():
             if art := self.get_articulation(convert_object_name(articulation)):
+                art.set_base_pose(articulation.root_pose)  # type: ignore
                 # set_qpos to update poses
                 art.set_qpos(articulation.qpos, full=True)  # type: ignore
             else: