diff --git a/metaworld/envs/sawyer_stick_pull_v3.py b/metaworld/envs/sawyer_stick_pull_v3.py index 087a01e4..3e957922 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 75f58303..102f681d 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]: