From 446ad187b134697b9779c74834df2a6c4d762f52 Mon Sep 17 00:00:00 2001
From: Richie Lo <Richie.Lo@dyson.com>
Date: Fri, 28 Jun 2024 18:14:16 +0100
Subject: [PATCH] add joint poses to misc observation

---
 rlbench/backend/scene.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/rlbench/backend/scene.py b/rlbench/backend/scene.py
index 64af9e897..96fc1ddd1 100644
--- 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