Fixing termination (Farama-Foundation#434)

reginald-mclean · Jul 28, 2023 · 844d76d · 844d76d
1 parent b4341e6
commit 844d76d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/metaworld/envs/mujoco/sawyer_xyz/sawyer_xyz_env.py b/metaworld/envs/mujoco/sawyer_xyz/sawyer_xyz_env.py
@@ -491,11 +491,11 @@ def step(self, action):
             dtype=np.float64,
         )
         reward, info = self.evaluate_state(self._last_stable_obs, action)
-        done = True if int(info["success"]) == 1 else False
+        # step will never return a terminal if there is a success
         return (
             np.array(self._last_stable_obs, dtype=np.float64),
             reward,
-            done,
+            False,
             False,
             info,
         )