From 53414fdbcc70412f299997516681b516bda63331 Mon Sep 17 00:00:00 2001 From: Wen Jiahao Date: Wed, 15 Jul 2026 20:49:30 +0800 Subject: [PATCH] Fix wrong stick indices --- metaworld/envs/sawyer_stick_pull_v3.py | 2 +- metaworld/envs/sawyer_stick_push_v3.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/metaworld/envs/sawyer_stick_pull_v3.py b/metaworld/envs/sawyer_stick_pull_v3.py index 087a01e49..3e9579228 100644 --- a/metaworld/envs/sawyer_stick_pull_v3.py +++ b/metaworld/envs/sawyer_stick_pull_v3.py @@ -144,7 +144,7 @@ def _set_obj_xyz(self, pos: npt.NDArray[Any]) -> None: qpos = self.data.qpos.flat.copy() qvel = self.data.qvel.flat.copy() qpos[16:18] = pos.copy() - qvel[16:18] = 0 + qvel[15:17] = 0 self.set_state(qpos, qvel) def reset_model(self) -> npt.NDArray[np.float64]: diff --git a/metaworld/envs/sawyer_stick_push_v3.py b/metaworld/envs/sawyer_stick_push_v3.py index 75f583033..102f681dd 100644 --- a/metaworld/envs/sawyer_stick_push_v3.py +++ b/metaworld/envs/sawyer_stick_push_v3.py @@ -141,7 +141,7 @@ def _set_obj_xyz(self, pos: npt.NDArray[Any]) -> None: qpos = self.data.qpos.flat.copy() qvel = self.data.qvel.flat.copy() qpos[16:18] = pos.copy() - qvel[16:18] = 0 + qvel[15:17] = 0 self.set_state(qpos, qvel) def reset_model(self) -> npt.NDArray[np.float64]: