diff --git a/predicators/envs/pybullet_env.py b/predicators/envs/pybullet_env.py index 5509e9c96c..73054b8031 100644 --- 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.