reformatted code

minigrid/core/world_object.py

@@ -25,7 +25,6 @@
 
 
 class WorldObj:
-
     """
     Base class for grid world objects
     """

minigrid/envs/babyai/core/levelgen.py

@@ -1,6 +1,7 @@
 """
 Copied and adapted from https://github.com/mila-iqia/babyai
 """
+
 from __future__ import annotations
 
 from minigrid.core.constants import COLOR_NAMES

minigrid/envs/babyai/core/roomgrid_level.py

@@ -1,6 +1,7 @@
 """
 Copied and adapted from https://github.com/mila-iqia/babyai
 """
+
 from __future__ import annotations
 
 from minigrid.core.roomgrid import RoomGrid

minigrid/envs/babyai/core/verifier.py

@@ -1,6 +1,7 @@
 """
 Copied and adapted from https://github.com/mila-iqia/babyai
 """
+
 from __future__ import annotations
 
 import os

minigrid/envs/babyai/goto.py

@@ -2,6 +2,7 @@
 Copied and adapted from https://github.com/mila-iqia/babyai.
 Levels described in the Baby AI ICLR 2019 submission, with the `Go to` instruction.
 """
+
 from __future__ import annotations
 
 from minigrid.envs.babyai.core.levelgen import LevelGen

minigrid/envs/babyai/open.py

@@ -2,6 +2,7 @@
 Copied and adapted from https://github.com/mila-iqia/babyai.
 Levels described in the Baby AI ICLR 2019 submission, with the `Open` instruction.
 """
+
 from __future__ import annotations
 
 from minigrid.core.constants import COLOR_NAMES

minigrid/envs/babyai/other.py

@@ -2,6 +2,7 @@
 Copied and adapted from https://github.com/mila-iqia/babyai.
 Levels described in the Baby AI ICLR 2019 submission, with different instructions than those in other files.
 """
+
 from __future__ import annotations
 
 from minigrid.envs.babyai.core.roomgrid_level import RoomGridLevel

minigrid/envs/babyai/pickup.py

@@ -2,6 +2,7 @@
 Copied and adapted from https://github.com/mila-iqia/babyai.
 Levels described in the Baby AI ICLR 2019 submission, with the `Pick up` instruction.
 """
+
 from __future__ import annotations
 
 from minigrid.envs.babyai.core.levelgen import LevelGen

minigrid/envs/babyai/putnext.py

@@ -2,6 +2,7 @@
 Copied and adapted from https://github.com/mila-iqia/babyai.
 Levels described in the Baby AI ICLR 2019 submission, with the `Put Next` instruction.
 """
+
 from __future__ import annotations
 
 from minigrid.envs.babyai.core.roomgrid_level import RoomGridLevel

minigrid/envs/babyai/unlock.py

@@ -2,6 +2,7 @@
 Copied and adapted from https://github.com/mila-iqia/babyai.
 Levels described in the Baby AI ICLR 2019 submission, with the `Unlock` instruction.
 """
+
 from __future__ import annotations
 
 from minigrid.core.constants import COLOR_NAMES

minigrid/envs/blockedunlockpickup.py

@@ -7,7 +7,6 @@
 
 
 class BlockedUnlockPickupEnv(RoomGrid):
-
     """
 
     ## Description

minigrid/envs/crossing.py

@@ -11,7 +11,6 @@
 
 
 class CrossingEnv(MiniGridEnv):
-
     """
     ## Description
 

minigrid/envs/distshift.py

@@ -7,7 +7,6 @@
 
 
 class DistShiftEnv(MiniGridEnv):
-
     """
     ## Description
 

minigrid/envs/doorkey.py

@@ -7,7 +7,6 @@
 
 
 class DoorKeyEnv(MiniGridEnv):
-
     """
     ## Description
 

minigrid/envs/fetch.py

@@ -8,7 +8,6 @@
 
 
 class FetchEnv(MiniGridEnv):
-
     """
     ## Description
 

minigrid/envs/fourrooms.py

@@ -7,7 +7,6 @@
 
 
 class FourRoomsEnv(MiniGridEnv):
-
     """
     ## Description
 

minigrid/envs/keycorridor.py

@@ -6,7 +6,6 @@
 
 
 class KeyCorridorEnv(RoomGrid):
-
     """
     ## Description
 

minigrid/envs/lavagap.py

@@ -9,7 +9,6 @@
 
 
 class LavaGapEnv(MiniGridEnv):
-
     """
     ## Description
 

minigrid/envs/lockedroom.py

@@ -22,7 +22,6 @@ def rand_pos(self, env):
 
 
 class LockedRoomEnv(MiniGridEnv):
-
     """
     ## Description
 

minigrid/envs/memory.py

@@ -10,7 +10,6 @@
 
 
 class MemoryEnv(MiniGridEnv):
-
     """
     ## Description
 

minigrid/envs/multiroom.py

@@ -16,7 +16,6 @@ def __init__(self, top, size, entryDoorPos, exitDoorPos):
 
 
 class MultiRoomEnv(MiniGridEnv):
-
     """
     ## Description
 

minigrid/envs/obstructedmaze.py

@@ -7,7 +7,6 @@
 
 
 class ObstructedMazeEnv(RoomGrid):
-
     """
     ## Description
 

minigrid/envs/putnear.py

@@ -8,7 +8,6 @@
 
 
 class PutNearEnv(MiniGridEnv):
-
     """
     ## Description
 

minigrid/envs/redbluedoors.py

@@ -7,7 +7,6 @@
 
 
 class RedBlueDoorEnv(MiniGridEnv):
-
     """
     ## Description
 

minigrid/envs/unlock.py

@@ -5,7 +5,6 @@
 
 
 class UnlockEnv(RoomGrid):
-
     """
     ## Description
 

minigrid/envs/unlockpickup.py

@@ -6,7 +6,6 @@
 
 
 class UnlockPickupEnv(RoomGrid):
-
     """
     ## Description
 

minigrid/envs/wfc/wfclogic/adjacency.py

@@ -1,4 +1,5 @@
 """Convert input data to adjacency information. Implementation based on https://github.com/ikarth/wfc_2019f"""
+
 from __future__ import annotations
 
 import numpy as np

minigrid/envs/wfc/wfclogic/control.py

@@ -1,4 +1,5 @@
 """Main WFC execution function. Implementation based on https://github.com/ikarth/wfc_2019f"""
+
 from __future__ import annotations
 
 import logging

minigrid/envs/wfc/wfclogic/solver.py

@@ -1,4 +1,5 @@
 """Wave Function Collapse solver. Implementation based on https://github.com/ikarth/wfc_2019f"""
+
 from __future__ import annotations
 
 import itertools

minigrid/envs/wfc/wfclogic/tiles.py

@@ -1,4 +1,5 @@
 """Breaks an image into consituant tiles. Implementation based on https://github.com/ikarth/wfc_2019f"""
+
 from __future__ import annotations
 
 import numpy as np

minigrid/envs/wfc/wfclogic/utilities.py

@@ -1,4 +1,5 @@
 """Utility data and functions for WFC. Implementation based on https://github.com/ikarth/wfc_2019f"""
+
 from __future__ import annotations
 
 import collections

minigrid/wrappers.py

@@ -1,8 +1,6 @@
 from __future__ import annotations
 
 import math
-import operator
-from functools import reduce
 from typing import Any
 
 import gymnasium as gym
@@ -596,7 +594,9 @@ def observation(self, obs):
             ), f"mission string too long ({len(mission)} chars)"
             mission = mission.lower()
 
-            str_array = np.zeros(shape=(self.maxStrLen, self.numCharCodes), dtype="uint8")
+            str_array = np.zeros(
+                shape=(self.maxStrLen, self.numCharCodes), dtype="uint8"
+            )
             # as `numCharCodes` < 255 then we can use `uint8`
 
             for idx, ch in enumerate(mission):

tests/test_wfc/test_wfc_adjacency.py

@@ -1,4 +1,5 @@
 """Convert input data to adjacency information"""
+
 from __future__ import annotations
 
 import numpy as np

tests/test_wfc/test_wfc_tiles.py

@@ -1,4 +1,5 @@
 """Breaks an image into consituant tiles."""
+
 from __future__ import annotations
 
 from minigrid.envs.wfc.wfclogic import tiles as wfc_tiles

tests/utils.py

@@ -1,4 +1,5 @@
 """Finds all the specs that we can test with"""
+
 from __future__ import annotations
 
 from importlib.util import find_spec