|
@@ -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'
|
|
|
+)
|