add joint poses to misc observation (#236)

Co-authored-by: Richie Lo <Richie.Lo@dyson.com>
stepjam · Jul 2, 2024 · 8fc31bf · 8fc31bf
1 parent 539138d
commit 8fc31bf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rlbench/backend/scene.py b/rlbench/backend/scene.py
@@ -556,4 +556,6 @@ def _get_cam_data(cam: VisionSensor, name: str):
         if self._joint_position_action is not None:
             # Store the actual requested joint positions during demo collection
             misc.update({"joint_position_action": self._joint_position_action})
+        joint_poses = [j.get_pose() for j in self.robot.arm.joints]
+        misc.update({'joint_poses': joint_poses})
         return misc