Update pick_cube.py

haosulab · Mar 3, 2024 · 9705f1c · 9705f1c
1 parent a3b4edd
commit 9705f1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mani_skill2/envs/tasks/pick_cube.py b/mani_skill2/envs/tasks/pick_cube.py
@@ -104,7 +104,7 @@ def _get_obs_extra(self, info: Dict):
         )
         if "state" in self.obs_mode:
             obs.update(
-                obs_pose=self.cube.pose.raw_pose,
+                obj_pose=self.cube.pose.raw_pose,
                 tcp_to_obj_pos=self.cube.pose.p - self.agent.tcp.pose.p,
                 obj_to_goal_pos=self.goal_site.pose.p - self.cube.pose.p,
             )