fix object position retrieval from environment using "cube_pos" naming.

perezjln · Aug 13, 2024 · cc8cc0d · cc8cc0d
1 parent 26eacbb
commit cc8cc0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/lerobotdataset_save.py b/examples/lerobotdataset_save.py
@@ -133,7 +133,7 @@ def process_args():
 
             state = torch.tensor(
                 np.concatenate(
-                    [np.array(obs_replay["arm_qpos"]), np.array(obs_replay["arm_qvel"]), np.array(obs_replay["object_qpos"])],
+                    [np.array(obs_replay["arm_qpos"]), np.array(obs_replay["arm_qvel"]), np.array(obs_replay["cube_pos"])],
                     axis=1,
                 )
             )