浏览代码

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 年之前
父节点
当前提交
bbcdb696ad
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 2 1
      gym_minigrid/minigrid.py

+ 2 - 1
gym_minigrid/minigrid.py

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