Browse Source

added kwargs to MiniGridEnv init

This will allow to pass arguments to reset() so that we can define the level by a teacher
vicliv 2 years ago
parent
commit
bbcdb696ad
1 changed files with 2 additions and 1 deletions
  1. 2 1
      gym_minigrid/minigrid.py

+ 2 - 1
gym_minigrid/minigrid.py

@@ -650,7 +650,8 @@ class MiniGridEnv(gym.Env):
         max_steps=100,
         see_through_walls=False,
         agent_view_size=7,
-        render_mode=None
+        render_mode=None,
+        **kwargs
     ):
         # Can't set both grid_size and width/height
         if grid_size: