add reset to held_obj_to_base_link

Learning-and-Intelligent-Systems · Jun 29, 2024 · 5bec0aa · 5bec0aa
1 parent 057e25d
commit 5bec0aa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/predicators/envs/pybullet_env.py b/predicators/envs/pybullet_env.py
@@ -197,6 +197,7 @@ def _reset_state(self, state: State) -> None:
             p.removeConstraint(self._held_constraint_id,
                                physicsClientId=self._physics_client_id)
             self._held_constraint_id = None
+        self._held_obj_to_base_link = None
         self._held_obj_id = None
 
         # Reset robot.