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, )