Explorar o código

Added more configs for GoToDoor environment

Maxime Chevalier-Boisvert %!s(int64=7) %!d(string=hai) anos
pai
achega
c75d108362
Modificáronse 1 ficheiros con 18 adicións e 0 borrados
  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'
+)