Removed waitEnds flag

gym_minigrid/envs/gotodoor.py

@@ -14,10 +14,7 @@ class GoToDoorEnv(MiniGridEnv):
         assert size >= 5
         super().__init__(gridSize=size, maxSteps=10*size)
 
-        self.reward_range = (-1000, 1000)
-
-        # Flag determining whether the wait action ends the episode
-        self.waitEnds = True
+        self.reward_range = (-1, 1)
 
     def _genGrid(self, width, height):
         # Create the grid
@@ -103,7 +100,6 @@ class GoToDoorEnv(MiniGridEnv):
         if action == self.actions.wait:
             if (ax == tx and abs(ay - ty) == 1) or (ay == ty and abs(ax - tx) == 1):
                 reward = 1
-            done = self.waitEnds
 
         obs = self._observation(obs)
 

gym_minigrid/envs/gotoobject.py

@@ -10,16 +10,13 @@ class GoToObjectEnv(MiniGridEnv):
     def __init__(
         self,
         size=6,
-        numObjs=2,
-        waitEnds=True
+        numObjs=2
     ):
         self.numObjs = numObjs
         super().__init__(gridSize=size, maxSteps=5*size)
 
         self.reward_range = (-1, 1)
 
-        self.waitEnds = waitEnds
-
     def _genGrid(self, width, height):
         assert width == height
         gridSz = width
@@ -112,7 +109,7 @@ class GoToObjectEnv(MiniGridEnv):
         if action == self.actions.wait:
             if abs(ax - tx) <= 1 and abs(ay - ty) <= 1:
                 reward = 1
-            done = self.waitEnds
+            done = True
 
         obs = self._observation(obs)