Преглед изворни кода

Added more configs for GoToDoor environment

Maxime Chevalier-Boisvert пре 7 година
родитељ
комит
c75d108362
1 измењених фајлова са 18 додато и 0 уклоњено
  1. 18 0
      gym_minigrid/envs/gotodoor.py

+ 18 - 0
gym_minigrid/envs/gotodoor.py

@@ -108,7 +108,25 @@ class GoToDoorEnv(MiniGridEnv):
 
         return obs, reward, done, info
 
+class GoToDoor8x8Env(GoToDoorEnv):
+    def __init__(self):
+        super().__init__(size=8)
+
+class GoToDoor6x6Env(GoToDoorEnv):
+    def __init__(self):
+        super().__init__(size=6)
+
 register(
     id='MiniGrid-GoToDoor-5x5-v0',
     entry_point='gym_minigrid.envs:GoToDoorEnv'
 )
+
+register(
+    id='MiniGrid-GoToDoor-6x6-v0',
+    entry_point='gym_minigrid.envs:GoToDoor6x6Env'
+)
+
+register(
+    id='MiniGrid-GoToDoor-8x8-v0',
+    entry_point='gym_minigrid.envs:GoToDoor8x8Env'
+)