Explorar o código

add **kwargs to reset function

saleml %!s(int64=2) %!d(string=hai) anos
pai
achega
3f3703355b
Modificáronse 1 ficheiros con 2 adicións e 2 borrados
  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