소스 검색

added kwargs to MiniGridEnv init

This will allow to pass arguments to reset() so that we can define the level by a teacher
vicliv 3 년 전
부모
커밋
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,
         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: