Run pre-commit

gym_minigrid/envs/__init__.py

@@ -10,10 +10,14 @@ from gym_minigrid.envs.gotodoor import GoToDoorEnv
 from gym_minigrid.envs.gotoobject import GoToObjectEnv
 from gym_minigrid.envs.keycorridor import KeyCorridorEnv
 from gym_minigrid.envs.lavagap import LavaGapEnv
-from gym_minigrid.envs.lockedroom import LockedRoomEnv, LockedRoom
+from gym_minigrid.envs.lockedroom import LockedRoom, LockedRoomEnv
 from gym_minigrid.envs.memory import MemoryEnv
-from gym_minigrid.envs.multiroom import MultiRoomEnv, MultiRoom
-from gym_minigrid.envs.obstructedmaze import ObstructedMazeEnv, ObstructedMaze_Full, ObstructedMaze_1Dlhb
+from gym_minigrid.envs.multiroom import MultiRoom, MultiRoomEnv
+from gym_minigrid.envs.obstructedmaze import (
+    ObstructedMaze_1Dlhb,
+    ObstructedMaze_Full,
+    ObstructedMazeEnv,
+)
 from gym_minigrid.envs.playground import PlaygroundEnv
 from gym_minigrid.envs.putnear import PutNearEnv
 from gym_minigrid.envs.redbluedoors import RedBlueDoorEnv

gym_minigrid/envs/blockedunlockpickup.py

@@ -49,6 +49,6 @@ class BlockedUnlockPickupEnv(RoomGrid):
 
 
 register(
-    id='MiniGrid-BlockedUnlockPickup-v0',
-    entry_point='gym_minigrid.envs.blockedunlockpickup:BlockedUnlockPickupEnv'
+    id="MiniGrid-BlockedUnlockPickup-v0",
+    entry_point="gym_minigrid.envs.blockedunlockpickup:BlockedUnlockPickupEnv",
 )

gym_minigrid/envs/crossing.py

@@ -86,49 +86,61 @@ class CrossingEnv(MiniGridEnv):
 
 
 register(
-    id='MiniGrid-LavaCrossingS9N1-v0',
-    entry_point='gym_minigrid.envs.crossing:CrossingEnv',
-    size=9, num_crossings=1
+    id="MiniGrid-LavaCrossingS9N1-v0",
+    entry_point="gym_minigrid.envs.crossing:CrossingEnv",
+    size=9,
+    num_crossings=1,
 )
 
 register(
-    id='MiniGrid-LavaCrossingS9N2-v0',
-    entry_point='gym_minigrid.envs.crossing:CrossingEnv',
-    size=9, num_crossings=2
+    id="MiniGrid-LavaCrossingS9N2-v0",
+    entry_point="gym_minigrid.envs.crossing:CrossingEnv",
+    size=9,
+    num_crossings=2,
 )
 
 register(
-    id='MiniGrid-LavaCrossingS9N3-v0',
-    entry_point='gym_minigrid.envs.crossing:CrossingEnv',
-    size=9, num_crossings=3
+    id="MiniGrid-LavaCrossingS9N3-v0",
+    entry_point="gym_minigrid.envs.crossing:CrossingEnv",
+    size=9,
+    num_crossings=3,
 )
 
 register(
-    id='MiniGrid-LavaCrossingS11N5-v0',
-    entry_point='gym_minigrid.envs.crossing:CrossingEnv',
-    size=11, num_crossings=5
+    id="MiniGrid-LavaCrossingS11N5-v0",
+    entry_point="gym_minigrid.envs.crossing:CrossingEnv",
+    size=11,
+    num_crossings=5,
 )
 
 register(
-    id='MiniGrid-SimpleCrossingS9N1-v0',
-    entry_point='gym_minigrid.envs.crossing:CrossingEnv',
-    size=9, num_crossings=1, obstacle_type=Wall
+    id="MiniGrid-SimpleCrossingS9N1-v0",
+    entry_point="gym_minigrid.envs.crossing:CrossingEnv",
+    size=9,
+    num_crossings=1,
+    obstacle_type=Wall,
 )
 
 register(
-    id='MiniGrid-SimpleCrossingS9N2-v0',
-    entry_point='gym_minigrid.envs.crossing:CrossingEnv',
-    size=9, num_crossings=2, obstacle_type=Wall
+    id="MiniGrid-SimpleCrossingS9N2-v0",
+    entry_point="gym_minigrid.envs.crossing:CrossingEnv",
+    size=9,
+    num_crossings=2,
+    obstacle_type=Wall,
 )
 
 register(
-    id='MiniGrid-SimpleCrossingS9N3-v0',
-    entry_point='gym_minigrid.envs.crossing:CrossingEnv',
-    size=9, num_crossings=3, obstacle_type=Wall
+    id="MiniGrid-SimpleCrossingS9N3-v0",
+    entry_point="gym_minigrid.envs.crossing:CrossingEnv",
+    size=9,
+    num_crossings=3,
+    obstacle_type=Wall,
 )
 
 register(
-    id='MiniGrid-SimpleCrossingS11N5-v0',
-    entry_point='gym_minigrid.envs.crossing:CrossingEnv',
-    size=11, num_crossings=5, obstacle_type=Wall
+    id="MiniGrid-SimpleCrossingS11N5-v0",
+    entry_point="gym_minigrid.envs.crossing:CrossingEnv",
+    size=11,
+    num_crossings=5,
+    obstacle_type=Wall,
 )

gym_minigrid/envs/distshift.py

@@ -56,13 +56,13 @@ class DistShiftEnv(MiniGridEnv):
 
 
 register(
-    id='MiniGrid-DistShift1-v0',
-    entry_point='gym_minigrid.envs.distshift:DistShiftEnv',
-strip2_row=2
+    id="MiniGrid-DistShift1-v0",
+    entry_point="gym_minigrid.envs.distshift:DistShiftEnv",
+    strip2_row=2,
 )
 
 register(
-    id='MiniGrid-DistShift2-v0',
-    entry_point='gym_minigrid.envs.distshift:DistShiftEnv',
-strip2_row=5
+    id="MiniGrid-DistShift2-v0",
+    entry_point="gym_minigrid.envs.distshift:DistShiftEnv",
+    strip2_row=5,
 )

gym_minigrid/envs/doorkey.py

@@ -41,25 +41,25 @@ class DoorKeyEnv(MiniGridEnv):
 
 
 register(
-    id='MiniGrid-DoorKey-5x5-v0',
-    entry_point='gym_minigrid.envs.doorkey:DoorKeyEnv',
-    size=5
+    id="MiniGrid-DoorKey-5x5-v0",
+    entry_point="gym_minigrid.envs.doorkey:DoorKeyEnv",
+    size=5,
 )
 
 register(
-    id='MiniGrid-DoorKey-6x6-v0',
-    entry_point='gym_minigrid.envs.doorkey:DoorKeyEnv',
-    size=6
+    id="MiniGrid-DoorKey-6x6-v0",
+    entry_point="gym_minigrid.envs.doorkey:DoorKeyEnv",
+    size=6,
 )
 
 register(
-    id='MiniGrid-DoorKey-8x8-v0',
-    entry_point='gym_minigrid.envs.doorkey:DoorKeyEnv',
-    size=8
+    id="MiniGrid-DoorKey-8x8-v0",
+    entry_point="gym_minigrid.envs.doorkey:DoorKeyEnv",
+    size=8,
 )
 
 register(
-    id='MiniGrid-DoorKey-16x16-v0',
-    entry_point='gym_minigrid.envs.doorkey:DoorKeyEnv',
-    size=16
+    id="MiniGrid-DoorKey-16x16-v0",
+    entry_point="gym_minigrid.envs.doorkey:DoorKeyEnv",
+    size=16,
 )

gym_minigrid/envs/dynamicobstacles.py

@@ -93,36 +93,43 @@ class DynamicObstaclesEnv(MiniGridEnv):
 
 
 register(
-    id='MiniGrid-Dynamic-Obstacles-5x5-v0',
-    entry_point='gym_minigrid.envs.dynamicobstacles:DynamicObstaclesEnv',
-    size=5, n_obstacles=2
+    id="MiniGrid-Dynamic-Obstacles-5x5-v0",
+    entry_point="gym_minigrid.envs.dynamicobstacles:DynamicObstaclesEnv",
+    size=5,
+    n_obstacles=2,
 )
 
 register(
-    id='MiniGrid-Dynamic-Obstacles-Random-5x5-v0',
-    entry_point='gym_minigrid.envs.dynamicobstacles:DynamicObstaclesEnv',
-    size=5, agent_start_pos=None, n_obstacles=2
+    id="MiniGrid-Dynamic-Obstacles-Random-5x5-v0",
+    entry_point="gym_minigrid.envs.dynamicobstacles:DynamicObstaclesEnv",
+    size=5,
+    agent_start_pos=None,
+    n_obstacles=2,
 )
 
 register(
-    id='MiniGrid-Dynamic-Obstacles-6x6-v0',
-    entry_point='gym_minigrid.envs.dynamicobstacles:DynamicObstaclesEnv',
-    size=6, n_obstacles=3
+    id="MiniGrid-Dynamic-Obstacles-6x6-v0",
+    entry_point="gym_minigrid.envs.dynamicobstacles:DynamicObstaclesEnv",
+    size=6,
+    n_obstacles=3,
 )
 
 register(
-    id='MiniGrid-Dynamic-Obstacles-Random-6x6-v0',
-    entry_point='gym_minigrid.envs.dynamicobstacles:DynamicObstaclesEnv',
-    size=6, agent_start_pos=None, n_obstacles=3
+    id="MiniGrid-Dynamic-Obstacles-Random-6x6-v0",
+    entry_point="gym_minigrid.envs.dynamicobstacles:DynamicObstaclesEnv",
+    size=6,
+    agent_start_pos=None,
+    n_obstacles=3,
 )
 
 register(
-    id='MiniGrid-Dynamic-Obstacles-8x8-v0',
-    entry_point='gym_minigrid.envs.dynamicobstacles:DynamicObstaclesEnv',
+    id="MiniGrid-Dynamic-Obstacles-8x8-v0",
+    entry_point="gym_minigrid.envs.dynamicobstacles:DynamicObstaclesEnv",
 )
 
 register(
-    id='MiniGrid-Dynamic-Obstacles-16x16-v0',
-    entry_point='gym_minigrid.envs.dynamicobstacles:DynamicObstaclesEnv',
-    size=16, n_obstacles=8
+    id="MiniGrid-Dynamic-Obstacles-16x16-v0",
+    entry_point="gym_minigrid.envs.dynamicobstacles:DynamicObstaclesEnv",
+    size=16,
+    n_obstacles=8,
 )

gym_minigrid/envs/empty.py

@@ -40,36 +40,36 @@ class EmptyEnv(MiniGridEnv):
 
 
 register(
-    id='MiniGrid-Empty-5x5-v0',
-    entry_point='gym_minigrid.envs.empty:EmptyEnv',
-    size=5
+    id="MiniGrid-Empty-5x5-v0", entry_point="gym_minigrid.envs.empty:EmptyEnv", size=5
 )
 
 register(
-    id='MiniGrid-Empty-Random-5x5-v0',
-    entry_point='gym_minigrid.envs.empty:EmptyEnv',
-    size=5, agent_start_pos=None
+    id="MiniGrid-Empty-Random-5x5-v0",
+    entry_point="gym_minigrid.envs.empty:EmptyEnv",
+    size=5,
+    agent_start_pos=None,
 )
 
 register(
-    id='MiniGrid-Empty-6x6-v0',
-    entry_point='gym_minigrid.envs.empty:EmptyEnv',
-size=6,
+    id="MiniGrid-Empty-6x6-v0",
+    entry_point="gym_minigrid.envs.empty:EmptyEnv",
+    size=6,
 )
 
 register(
-    id='MiniGrid-Empty-Random-6x6-v0',
-    entry_point='gym_minigrid.envs.empty:EmptyEnv',
-size=6, agent_start_pos=None
+    id="MiniGrid-Empty-Random-6x6-v0",
+    entry_point="gym_minigrid.envs.empty:EmptyEnv",
+    size=6,
+    agent_start_pos=None,
 )
 
 register(
-    id='MiniGrid-Empty-8x8-v0',
-    entry_point='gym_minigrid.envs.empty:EmptyEnv',
+    id="MiniGrid-Empty-8x8-v0",
+    entry_point="gym_minigrid.envs.empty:EmptyEnv",
 )
 
 register(
-    id='MiniGrid-Empty-16x16-v0',
-    entry_point='gym_minigrid.envs.empty:EmptyEnv',
-size=16,
+    id="MiniGrid-Empty-16x16-v0",
+    entry_point="gym_minigrid.envs.empty:EmptyEnv",
+    size=16,
 )

gym_minigrid/envs/fetch.py

@@ -87,18 +87,17 @@ class FetchEnv(MiniGridEnv):
 
 
 register(
-    id='MiniGrid-Fetch-5x5-N2-v0',
-    entry_point='gym_minigrid.envs.fetch:FetchEnv',
-    size=5, numObjs=2
+    id="MiniGrid-Fetch-5x5-N2-v0",
+    entry_point="gym_minigrid.envs.fetch:FetchEnv",
+    size=5,
+    numObjs=2,
 )
 
 register(
-    id='MiniGrid-Fetch-6x6-N2-v0',
-    entry_point='gym_minigrid.envs.fetch:FetchEnv',
-    size=6, numObjs=2
+    id="MiniGrid-Fetch-6x6-N2-v0",
+    entry_point="gym_minigrid.envs.fetch:FetchEnv",
+    size=6,
+    numObjs=2,
 )
 
-register(
-    id='MiniGrid-Fetch-8x8-N3-v0',
-    entry_point='gym_minigrid.envs.fetch:FetchEnv'
-)
+register(id="MiniGrid-Fetch-8x8-N3-v0", entry_point="gym_minigrid.envs.fetch:FetchEnv")

gym_minigrid/envs/fourrooms.py

@@ -73,6 +73,5 @@ class FourRoomsEnv(MiniGridEnv):
 
 
 register(
-    id='MiniGrid-FourRooms-v0',
-    entry_point='gym_minigrid.envs.fourrooms:FourRoomsEnv'
+    id="MiniGrid-FourRooms-v0", entry_point="gym_minigrid.envs.fourrooms:FourRoomsEnv"
 )

gym_minigrid/envs/gotodoor.py

@@ -81,18 +81,17 @@ class GoToDoorEnv(MiniGridEnv):
 
 
 register(
-    id='MiniGrid-GoToDoor-5x5-v0',
-    entry_point='gym_minigrid.envs.gotodoor:GoToDoorEnv'
+    id="MiniGrid-GoToDoor-5x5-v0", entry_point="gym_minigrid.envs.gotodoor:GoToDoorEnv"
 )
 
 register(
-    id='MiniGrid-GoToDoor-6x6-v0',
-    entry_point='gym_minigrid.envs.gotodoor:GoToDoorEnv',
-    size=6
+    id="MiniGrid-GoToDoor-6x6-v0",
+    entry_point="gym_minigrid.envs.gotodoor:GoToDoorEnv",
+    size=6,
 )
 
 register(
-    id='MiniGrid-GoToDoor-8x8-v0',
-    entry_point='gym_minigrid.envs.gotodoor:GoToDoorEnv',
-    size=8
+    id="MiniGrid-GoToDoor-8x8-v0",
+    entry_point="gym_minigrid.envs.gotodoor:GoToDoorEnv",
+    size=8,
 )

gym_minigrid/envs/gotoobject.py

@@ -83,12 +83,13 @@ class GoToObjectEnv(MiniGridEnv):
 
 
 register(
-    id='MiniGrid-GoToObject-6x6-N2-v0',
-    entry_point='gym_minigrid.envs.gotoobject:GoToObjectEnv'
+    id="MiniGrid-GoToObject-6x6-N2-v0",
+    entry_point="gym_minigrid.envs.gotoobject:GoToObjectEnv",
 )
 
 register(
-    id='MiniGrid-GoToObject-8x8-N2-v0',
-    entry_point='gym_minigrid.envs.gotoobject:GoToObjectEnv',
-    size=8, numObjs=2
+    id="MiniGrid-GoToObject-8x8-N2-v0",
+    entry_point="gym_minigrid.envs.gotoobject:GoToObjectEnv",
+    size=8,
+    numObjs=2,
 )

gym_minigrid/envs/keycorridor.py

@@ -55,43 +55,43 @@ class KeyCorridorEnv(RoomGrid):
 
 
 register(
-    id='MiniGrid-KeyCorridorS3R1-v0',
-    entry_point='gym_minigrid.envs.keycorridor:KeyCorridorEnv',
+    id="MiniGrid-KeyCorridorS3R1-v0",
+    entry_point="gym_minigrid.envs.keycorridor:KeyCorridorEnv",
     room_size=3,
     num_rows=1,
 )
 
 register(
-    id='MiniGrid-KeyCorridorS3R2-v0',
-    entry_point='gym_minigrid.envs.keycorridor:KeyCorridorEnv',
+    id="MiniGrid-KeyCorridorS3R2-v0",
+    entry_point="gym_minigrid.envs.keycorridor:KeyCorridorEnv",
     room_size=3,
     num_rows=2,
 )
 
 register(
-    id='MiniGrid-KeyCorridorS3R3-v0',
-    entry_point='gym_minigrid.envs.keycorridor:KeyCorridorEnv',
+    id="MiniGrid-KeyCorridorS3R3-v0",
+    entry_point="gym_minigrid.envs.keycorridor:KeyCorridorEnv",
     room_size=3,
     num_rows=3,
 )
 
 register(
-    id='MiniGrid-KeyCorridorS4R3-v0',
-    entry_point='gym_minigrid.envs.keycorridor:KeyCorridorEnv',
+    id="MiniGrid-KeyCorridorS4R3-v0",
+    entry_point="gym_minigrid.envs.keycorridor:KeyCorridorEnv",
     room_size=4,
     num_rows=3,
 )
 
 register(
-    id='MiniGrid-KeyCorridorS5R3-v0',
-    entry_point='gym_minigrid.envs.keycorridor:KeyCorridorEnv',
+    id="MiniGrid-KeyCorridorS5R3-v0",
+    entry_point="gym_minigrid.envs.keycorridor:KeyCorridorEnv",
     room_size=5,
     num_rows=3,
 )
 
 register(
-    id='MiniGrid-KeyCorridorS6R3-v0',
-    entry_point='gym_minigrid.envs.keycorridor:KeyCorridorEnv',
+    id="MiniGrid-KeyCorridorS6R3-v0",
+    entry_point="gym_minigrid.envs.keycorridor:KeyCorridorEnv",
     room_size=6,
     num_rows=3,
 )

gym_minigrid/envs/lavagap.py

@@ -59,19 +59,19 @@ class LavaGapEnv(MiniGridEnv):
 
 
 register(
-    id='MiniGrid-LavaGapS5-v0',
-    entry_point='gym_minigrid.envs.lavagap:LavaGapEnv',
-    size=5
+    id="MiniGrid-LavaGapS5-v0",
+    entry_point="gym_minigrid.envs.lavagap:LavaGapEnv",
+    size=5,
 )
 
 register(
-    id='MiniGrid-LavaGapS6-v0',
-    entry_point='gym_minigrid.envs.lavagap:LavaGapEnv',
-    size=6
+    id="MiniGrid-LavaGapS6-v0",
+    entry_point="gym_minigrid.envs.lavagap:LavaGapEnv",
+    size=6,
 )
 
 register(
-    id='MiniGrid-LavaGapS7-v0',
-    entry_point='gym_minigrid.envs.lavagap:LavaGapEnv',
-    size=7
+    id="MiniGrid-LavaGapS7-v0",
+    entry_point="gym_minigrid.envs.lavagap:LavaGapEnv",
+    size=7,
 )

gym_minigrid/envs/lockedroom.py

@@ -1,12 +1,9 @@
 from gym_minigrid.minigrid import COLOR_NAMES, Door, Goal, Grid, Key, MiniGridEnv, Wall
 from gym_minigrid.register import register
 
+
 class LockedRoom:
-    def __init__(self,
-        top,
-        size,
-        doorPos
-    ):
+    def __init__(self, top, size, doorPos):
         self.top = top
         self.size = size
         self.doorPos = doorPos
@@ -59,16 +56,10 @@ class LockedRoomEnv(MiniGridEnv):
 
             roomW = lWallIdx + 1
             roomH = height // 3 + 1
-            self.rooms.append(LockedRoom(
-                (0, j),
-                (roomW, roomH),
-                (lWallIdx, j + 3)
-            ))
-            self.rooms.append(LockedRoom(
-                (rWallIdx, j),
-                (roomW, roomH),
-                (rWallIdx, j + 3)
-            ))
+            self.rooms.append(LockedRoom((0, j), (roomW, roomH), (lWallIdx, j + 3)))
+            self.rooms.append(
+                LockedRoom((rWallIdx, j), (roomW, roomH), (rWallIdx, j + 3))
+            )
 
         # Choose one random room to be locked
         lockedRoom = self._rand_elem(self.rooms)
@@ -111,7 +102,8 @@ class LockedRoomEnv(MiniGridEnv):
         obs, reward, done, info = MiniGridEnv.step(self, action)
         return obs, reward, done, info
 
+
 register(
-    id='MiniGrid-LockedRoom-v0',
-    entry_point='gym_minigrid.envs.lockedroom:LockedRoomEnv'
+    id="MiniGrid-LockedRoom-v0",
+    entry_point="gym_minigrid.envs.lockedroom:LockedRoomEnv",
 )

gym_minigrid/envs/memory.py

@@ -97,39 +97,37 @@ class MemoryEnv(MiniGridEnv):
 
 
 register(
-    id='MiniGrid-MemoryS17Random-v0',
-    entry_point='gym_minigrid.envs.memory:MemoryEnv',
-    size=17, random_length=True
+    id="MiniGrid-MemoryS17Random-v0",
+    entry_point="gym_minigrid.envs.memory:MemoryEnv",
+    size=17,
+    random_length=True,
 )
 
 register(
-    id='MiniGrid-MemoryS13Random-v0',
-    entry_point='gym_minigrid.envs.memory:MemoryEnv',
-    size=13, random_length=True
+    id="MiniGrid-MemoryS13Random-v0",
+    entry_point="gym_minigrid.envs.memory:MemoryEnv",
+    size=13,
+    random_length=True,
 )
 
 
 register(
-    id='MiniGrid-MemoryS13-v0',
-    entry_point='gym_minigrid.envs.memory:MemoryEnv',
-    size=13
+    id="MiniGrid-MemoryS13-v0",
+    entry_point="gym_minigrid.envs.memory:MemoryEnv",
+    size=13,
 )
 
 
 register(
-    id='MiniGrid-MemoryS11-v0',
-    entry_point='gym_minigrid.envs.memory:MemoryEnv',
-    size=11
+    id="MiniGrid-MemoryS11-v0",
+    entry_point="gym_minigrid.envs.memory:MemoryEnv",
+    size=11,
 )
 
 register(
-    id='MiniGrid-MemoryS9-v0',
-    entry_point='gym_minigrid.envs.memory:MemoryEnv',
-    size=9
+    id="MiniGrid-MemoryS9-v0", entry_point="gym_minigrid.envs.memory:MemoryEnv", size=9
 )
 
 register(
-    id='MiniGrid-MemoryS7-v0',
-    entry_point='gym_minigrid.envs.memory:MemoryEnv',
-    size=7
+    id="MiniGrid-MemoryS7-v0", entry_point="gym_minigrid.envs.memory:MemoryEnv", size=7
 )

gym_minigrid/envs/multiroom.py

@@ -1,13 +1,9 @@
 from gym_minigrid.minigrid import COLOR_NAMES, Door, Goal, Grid, MiniGridEnv, Wall
 from gym_minigrid.register import register
 
+
 class MultiRoom:
-    def __init__(self,
-        top,
-        size,
-        entryDoorPos,
-        exitDoorPos
-    ):
+    def __init__(self, top, size, entryDoorPos, exitDoorPos):
         self.top = top
         self.size = size
         self.entryDoorPos = entryDoorPos
@@ -157,12 +153,7 @@ class MultiRoomEnv(MiniGridEnv):
                 return False
 
         # Add this room to the list
-        roomList.append(MultiRoom(
-            (topX, topY),
-            (sizeX, sizeY),
-            entryDoorPos,
-            None
-        ))
+        roomList.append(MultiRoom((topX, topY), (sizeX, sizeY), entryDoorPos, None))
 
         # If this was the last room, stop
         if numLeft == 1:
@@ -210,24 +201,24 @@ class MultiRoomEnv(MiniGridEnv):
 
 
 register(
-    id='MiniGrid-MultiRoom-N2-S4-v0',
-    entry_point='gym_minigrid.envs.multiroom:MultiRoomEnv',
+    id="MiniGrid-MultiRoom-N2-S4-v0",
+    entry_point="gym_minigrid.envs.multiroom:MultiRoomEnv",
     minNumRooms=2,
     maxNumRooms=2,
-    maxRoomSize=4
+    maxRoomSize=4,
 )
 
 register(
-    id='MiniGrid-MultiRoom-N4-S5-v0',
-    entry_point='gym_minigrid.envs.multiroom:MultiRoomEnv',
+    id="MiniGrid-MultiRoom-N4-S5-v0",
+    entry_point="gym_minigrid.envs.multiroom:MultiRoomEnv",
     minNumRooms=4,
     maxNumRooms=4,
-    maxRoomSize=5
+    maxRoomSize=5,
 )
 
 register(
-    id='MiniGrid-MultiRoom-N6-v0',
-    entry_point='gym_minigrid.envs.multiroom:MultiRoomEnv',
+    id="MiniGrid-MultiRoom-N6-v0",
+    entry_point="gym_minigrid.envs.multiroom:MultiRoomEnv",
     minNumRooms=6,
-    maxNumRooms=6
+    maxNumRooms=6,
 )

gym_minigrid/envs/obstructedmaze.py

@@ -181,60 +181,76 @@ class ObstructedMaze_2Dlhb(ObstructedMaze_Full):
         super().__init__((2, 1), True, True, 1, 4, **kwargs)
 
 
-
 register(
     id="MiniGrid-ObstructedMaze-1Dl-v0",
     entry_point="gym_minigrid.envs.obstructedmaze:ObstructedMaze_1Dlhb",
-    key_in_box=False, blocked=False
+    key_in_box=False,
+    blocked=False,
 )
 
 register(
     id="MiniGrid-ObstructedMaze-1Dlh-v0",
     entry_point="gym_minigrid.envs.obstructedmaze:ObstructedMaze_1Dlhb",
-    key_in_box=True, blocked=False
+    key_in_box=True,
+    blocked=False,
 )
 
 register(
     id="MiniGrid-ObstructedMaze-1Dlhb-v0",
-    entry_point="gym_minigrid.envs.obstructedmaze:ObstructedMaze_1Dlhb"
+    entry_point="gym_minigrid.envs.obstructedmaze:ObstructedMaze_1Dlhb",
 )
 
 register(
     id="MiniGrid-ObstructedMaze-2Dl-v0",
     entry_point="gym_minigrid.envs.obstructedmaze:ObstructedMaze_Full",
-    agent_room=(2, 1), key_in_box=False, blocked=False,
-    num_quarters=1, num_rooms_visited=4,
+    agent_room=(2, 1),
+    key_in_box=False,
+    blocked=False,
+    num_quarters=1,
+    num_rooms_visited=4,
 )
 
 register(
     id="MiniGrid-ObstructedMaze-2Dlh-v0",
     entry_point="gym_minigrid.envs.obstructedmaze:ObstructedMaze_Full",
-    agent_room=(2, 1), key_in_box=True, blocked=False,
-    num_quarters=1, num_rooms_visited=4,
+    agent_room=(2, 1),
+    key_in_box=True,
+    blocked=False,
+    num_quarters=1,
+    num_rooms_visited=4,
 )
 
 register(
     id="MiniGrid-ObstructedMaze-2Dlhb-v0",
     entry_point="gym_minigrid.envs.obstructedmaze:ObstructedMaze_Full",
-    agent_room=(2, 1), key_in_box=True, blocked=True,
-    num_quarters=1, num_rooms_visited=4,
+    agent_room=(2, 1),
+    key_in_box=True,
+    blocked=True,
+    num_quarters=1,
+    num_rooms_visited=4,
 )
 
 register(
     id="MiniGrid-ObstructedMaze-1Q-v0",
     entry_point="gym_minigrid.envs.obstructedmaze:ObstructedMaze_Full",
-    agent_room=(1, 1), key_in_box=True, blocked=True,
-    num_quarters=1, num_rooms_visited=5,
+    agent_room=(1, 1),
+    key_in_box=True,
+    blocked=True,
+    num_quarters=1,
+    num_rooms_visited=5,
 )
 
 register(
     id="MiniGrid-ObstructedMaze-2Q-v0",
     entry_point="gym_minigrid.envs.obstructedmaze:ObstructedMaze_Full",
-    agent_room=(2, 1), key_in_box=True, blocked=True,
-    num_quarters=2, num_rooms_visited=11,
+    agent_room=(2, 1),
+    key_in_box=True,
+    blocked=True,
+    num_quarters=2,
+    num_rooms_visited=11,
 )
 
 register(
     id="MiniGrid-ObstructedMaze-Full-v0",
-    entry_point="gym_minigrid.envs.obstructedmaze:ObstructedMaze_Full"
-)
+    entry_point="gym_minigrid.envs.obstructedmaze:ObstructedMaze_Full",
+)

gym_minigrid/envs/playground.py

@@ -1,6 +1,7 @@
 from gym_minigrid.minigrid import COLOR_NAMES, Ball, Box, Door, Grid, Key, MiniGridEnv
 from gym_minigrid.register import register
 
+
 class PlaygroundEnv(MiniGridEnv):
     """
     Environment with multiple rooms and random objects.
@@ -70,7 +71,8 @@ class PlaygroundEnv(MiniGridEnv):
         obs, reward, done, info = super().step(action)
         return obs, reward, done, info
 
+
 register(
-    id='MiniGrid-Playground-v0',
-    entry_point='gym_minigrid.envs.playground:PlaygroundEnv'
+    id="MiniGrid-Playground-v0",
+    entry_point="gym_minigrid.envs.playground:PlaygroundEnv",
 )

gym_minigrid/envs/putnear.py

@@ -114,12 +114,12 @@ class PutNearEnv(MiniGridEnv):
 
 
 register(
-    id='MiniGrid-PutNear-6x6-N2-v0',
-    entry_point='gym_minigrid.envs.putnear:PutNearEnv'
+    id="MiniGrid-PutNear-6x6-N2-v0", entry_point="gym_minigrid.envs.putnear:PutNearEnv"
 )
 
 register(
-    id='MiniGrid-PutNear-8x8-N3-v0',
-    entry_point='gym_minigrid.envs.putnear:PutNearEnv',
-    size=8, numObjs=3
+    id="MiniGrid-PutNear-8x8-N3-v0",
+    entry_point="gym_minigrid.envs.putnear:PutNearEnv",
+    size=8,
+    numObjs=3,
 )

gym_minigrid/envs/redbluedoors.py

@@ -66,12 +66,12 @@ class RedBlueDoorEnv(MiniGridEnv):
 
 
 register(
-    id='MiniGrid-RedBlueDoors-6x6-v0',
-    entry_point='gym_minigrid.envs.redbluedoors:RedBlueDoorEnv',
-    size=6
+    id="MiniGrid-RedBlueDoors-6x6-v0",
+    entry_point="gym_minigrid.envs.redbluedoors:RedBlueDoorEnv",
+    size=6,
 )
 
 register(
-    id='MiniGrid-RedBlueDoors-8x8-v0',
-    entry_point='gym_minigrid.envs.redbluedoors:RedBlueDoorEnv'
+    id="MiniGrid-RedBlueDoors-8x8-v0",
+    entry_point="gym_minigrid.envs.redbluedoors:RedBlueDoorEnv",
 )

gym_minigrid/envs/unlock.py

@@ -40,7 +40,5 @@ class UnlockEnv(RoomGrid):
 
         return obs, reward, done, info
 
-register(
-    id='MiniGrid-Unlock-v0',
-    entry_point='gym_minigrid.envs.unlock:UnlockEnv'
-)
+
+register(id="MiniGrid-Unlock-v0", entry_point="gym_minigrid.envs.unlock:UnlockEnv")

gym_minigrid/envs/unlockpickup.py

@@ -42,7 +42,8 @@ class UnlockPickupEnv(RoomGrid):
 
         return obs, reward, done, info
 
+
 register(
-    id='MiniGrid-UnlockPickup-v0',
-    entry_point='gym_minigrid.envs.unlockpickup:UnlockPickupEnv'
+    id="MiniGrid-UnlockPickup-v0",
+    entry_point="gym_minigrid.envs.unlockpickup:UnlockPickupEnv",
 )

gym_minigrid/register.py

@@ -3,21 +3,13 @@ from gym.envs.registration import register as gym_register
 env_list = []
 
 
-def register(
-    id,
-    entry_point,
-    reward_threshold=0.95,
-    **kwargs
-):
+def register(id, entry_point, reward_threshold=0.95, **kwargs):
     assert id.startswith("MiniGrid-")
     assert id not in env_list
 
     # Register the environment with OpenAI gym
     gym_register(
-        id=id,
-        entry_point=entry_point,
-        reward_threshold=reward_threshold,
-        kwargs=kwargs
+        id=id, entry_point=entry_point, reward_threshold=reward_threshold, kwargs=kwargs
     )
 
     # Add the environment to the set