Fix issue when generating non-square DoorKey environments (Farama-Fou…

…ndation#427)
rbankosegger · Apr 22, 2024 · 9b8f7b7 · 9b8f7b7
1 parent 617e490
commit 9b8f7b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/minigrid/envs/doorkey.py b/minigrid/envs/doorkey.py
@@ -91,7 +91,7 @@ def _gen_grid(self, width, height):
  self.place_agent(size=(splitIdx, height))
 
  # Place a door in the wall
- doorIdx = self._rand_int(1, width - 2)
+ doorIdx = self._rand_int(1, height - 2)
  self.put_obj(Door("yellow", is_locked=True), splitIdx, doorIdx)
 
  # Place a yellow key on the left side