From 9177e8b96c4eefc671ce00e866c50b4636bd6da9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bart=C5=82omiej=20Cupia=C5=82?=
 <bartlomiej.cupial@gmail.com>
Date: Sat, 30 Nov 2024 08:18:05 +0100
Subject: [PATCH] configurable multiroom

---
 minigrid/envs/multiroom.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/minigrid/envs/multiroom.py b/minigrid/envs/multiroom.py
index b998a492e..3c38365ed 100644
--- 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,
         )