소스 검색

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