Browse Source

Added configuration for MultiRoom env

Maxime Chevalier-Boisvert 6 years ago
parent
commit
999599a412
2 changed files with 16 additions and 4 deletions
  1. 1 0
      README.md
  2. 15 4
      gym_minigrid/envs/multiroom.py

+ 1 - 0
README.md

@@ -181,6 +181,7 @@ useful to experiment with curiosity or curriculum learning.
 
 Registered configurations:
 - `MiniGrid-MultiRoom-N2-S4-v0` (two small rooms)
+- `MiniGrid-MultiRoom-N4-S5-v0` (four rooms)
 - `MiniGrid-MultiRoom-N6-v0` (six rooms)
 
 <p align="center">

+ 15 - 4
gym_minigrid/envs/multiroom.py

@@ -244,6 +244,14 @@ class MultiRoomEnvN2S4(MultiRoomEnv):
             maxRoomSize=4
         )
 
+class MultiRoomEnvN4S5(MultiRoomEnv):
+    def __init__(self):
+        super().__init__(
+            minNumRooms=4,
+            maxNumRooms=4,
+            maxRoomSize=5
+        )
+
 class MultiRoomEnvN6(MultiRoomEnv):
     def __init__(self):
         super().__init__(
@@ -253,12 +261,15 @@ class MultiRoomEnvN6(MultiRoomEnv):
 
 register(
     id='MiniGrid-MultiRoom-N2-S4-v0',
-    entry_point='gym_minigrid.envs:MultiRoomEnvN2S4',
-    reward_threshold=1000.0
+    entry_point='gym_minigrid.envs:MultiRoomEnvN2S4'
+)
+
+register(
+    id='MiniGrid-MultiRoom-N4-S5-v0',
+    entry_point='gym_minigrid.envs:MultiRoomEnvN4S5'
 )
 
 register(
     id='MiniGrid-MultiRoom-N6-v0',
-    entry_point='gym_minigrid.envs:MultiRoomEnvN6',
-    reward_threshold=1000.0
+    entry_point='gym_minigrid.envs:MultiRoomEnvN6'
 )