configurable multiroom

Farama-Foundation · Nov 30, 2024 · f1c9b42 · f1c9b42
1 parent a6cfc0e
commit f1c9b42
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/minigrid/envs/multiroom.py b/minigrid/envs/multiroom.py
@@ -76,6 +76,8 @@ def __init__(
         minNumRooms,
         maxNumRooms,
         maxRoomSize=10,
+        width=25,
+        height=25,
         max_steps: int | None = None,
         **kwargs,
     ):
@@ -91,15 +93,13 @@ def __init__(
 
         mission_space = MissionSpace(mission_func=self._gen_mission)
 
-        self.size = 25
-
         if max_steps is None:
             max_steps = maxNumRooms * 20
 
         super().__init__(
             mission_space=mission_space,
-            width=self.size,
-            height=self.size,
+            width=width,
+            height=height,
             max_steps=max_steps,
             **kwargs,
         )