Fix slicing issue in push when random_init is False (#368)

Fixing #367
Farama-Foundation · Apr 19, 2022 · b0b66d1 · b0b66d1
1 parent 29fe5d6
commit b0b66d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metaworld/envs/mujoco/sawyer_xyz/v2/sawyer_push_v2.py b/metaworld/envs/mujoco/sawyer_xyz/v2/sawyer_push_v2.py
@@ -119,7 +119,7 @@ def fix_extreme_obj_pos(self, orig_init_pos):
     def reset_model(self):
         self._reset_hand()
         self._target_pos = self.goal.copy()
-        self.obj_init_pos = self.fix_extreme_obj_pos(self.init_config['obj_init_pos'])
+        self.obj_init_pos = np.array(self.fix_extreme_obj_pos(self.init_config['obj_init_pos']))
         self.obj_init_angle = self.init_config['obj_init_angle']
 
         if self.random_init: