Browse Source

add **kwargs to reset function

saleml 2 years ago
parent
commit
3f3703355b
1 changed files with 2 additions and 2 deletions
  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