Переглянути джерело

add **kwargs to reset function

saleml 2 роки тому
батько
коміт
3f3703355b
1 змінених файлів з 2 додано та 2 видалено
  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