ソースを参照

allow passing kwargs to EmptyEnvNxN (#83)

(agent_start_pos and agent_dir)
rosea-tf 5 年 前
コミット
ded28f95ae
1 ファイル変更6 行追加6 行削除
  1. 6 6
      gym_minigrid/envs/empty.py

+ 6 - 6
gym_minigrid/envs/empty.py

@@ -42,24 +42,24 @@ class EmptyEnv(MiniGridEnv):
         self.mission = "get to the green goal square"
 
 class EmptyEnv5x5(EmptyEnv):
-    def __init__(self):
-        super().__init__(size=5)
+    def __init__(self, **kwargs):
+        super().__init__(size=5, **kwargs)
 
 class EmptyRandomEnv5x5(EmptyEnv):
     def __init__(self):
         super().__init__(size=5, agent_start_pos=None)
 
 class EmptyEnv6x6(EmptyEnv):
-    def __init__(self):
-        super().__init__(size=6)
+    def __init__(self, **kwargs):
+        super().__init__(size=6, **kwargs)
 
 class EmptyRandomEnv6x6(EmptyEnv):
     def __init__(self):
         super().__init__(size=6, agent_start_pos=None)
 
 class EmptyEnv16x16(EmptyEnv):
-    def __init__(self):
-        super().__init__(size=16)
+    def __init__(self, **kwargs):
+        super().__init__(size=16, **kwargs)
 
 register(
     id='MiniGrid-Empty-5x5-v0',