Forráskód Böngészése

Changed reward range for fetch environment

Maxime Chevalier-Boisvert 7 éve
szülő
commit
32535cc145
1 módosított fájl, 2 hozzáadás és 1 törlés
  1. 2 1
      gym_minigrid/envs/fetch.py

+ 2 - 1
gym_minigrid/envs/fetch.py

@@ -14,6 +14,7 @@ class FetchEnv(MiniGridEnv):
     ):
         self.numObjs = numObjs
         super().__init__(gridSize=size, maxSteps=5*size)
+        self.reward_range = (-1000, 1000)
 
     def _genGrid(self, width, height):
         assert width == height
@@ -104,7 +105,7 @@ class FetchEnv(MiniGridEnv):
                 reward = 1000 - self.stepCount
                 done = True
             else:
-                reward = -1
+                reward = -1000
                 done = True
 
         obs = self._observation(obs)