Browse Source

In the new gym api, reset takes two extra arguments. This removes some warnings.

saleml 2 years ago
parent
commit
22b50565dd
1 changed files with 2 additions and 1 deletions
  1. 2 1
      gym_minigrid/minigrid.py

+ 2 - 1
gym_minigrid/minigrid.py

@@ -728,7 +728,8 @@ class MiniGridEnv(gym.Env):
         # Initialize the state
         self.reset()
 
-    def reset(self):
+    def reset(self, *, seed=None, return_info=False, options=None):
+        super().reset(seed=seed)
         # Current position and direction of the agent
         self.agent_pos = None
         self.agent_dir = None