Added MiniGrid-MultiRoom-N2-S4-v0 environment

README.md

@@ -132,6 +132,7 @@ useful to experiment with curiosity or curriculum learning.
 ### Multi-room environment
 
 Registered configurations:
+- `MiniGrid-MultiRoom-N2-S4-v0`
 - `MiniGrid-MultiRoom-N6-v0`
 
 <p align="center">

gym_minigrid/envs/multiroom.py

@@ -252,14 +252,28 @@ class MultiRoomEnv(MiniGridEnv):
 
         return True
 
+class MultiRoomEnvN2S4(MultiRoomEnv):
+    def __init__(self):
+        super().__init__(
+            minNumRooms=2,
+            maxNumRooms=2,
+            maxRoomSize=4
+        )
+
 class MultiRoomEnvN6(MultiRoomEnv):
     def __init__(self):
-        super(MultiRoomEnvN6, self).__init__(
+        super().__init__(
             minNumRooms=6,
             maxNumRooms=6
         )
 
 register(
+    id='MiniGrid-MultiRoom-N2-S4-v0',
+    entry_point='gym_minigrid.envs:MultiRoomEnvN2S4',
+    reward_threshold=1000.0
+)
+
+register(
     id='MiniGrid-MultiRoom-N6-v0',
     entry_point='gym_minigrid.envs:MultiRoomEnvN6',
     reward_threshold=1000.0