Parcourir la source

add **kwargs to reset function

saleml il y a 2 ans
Parent
commit
3f3703355b
1 fichiers modifiés avec 2 ajouts et 2 suppressions
  1. 2 2
      run_tests.py

+ 2 - 2
run_tests.py

@@ -157,8 +157,8 @@ class EmptyEnvWithExtraObs(gym_minigrid.envs.EmptyEnv5x5):
             dtype=np.uint
         )
 
-    def reset(self):
-        obs = super().reset()
+    def reset(self, **kwargs):
+        obs = super().reset(**kwargs)
         obs['size'] = np.array([self.width, self.height])
         return obs