precommit

Farama-Foundation · Oct 31, 2024 · cf73dd1 · cf73dd1
1 parent ad8a66b
commit cf73dd1
Show file tree

Hide file tree

Showing 5 changed files with 427 additions and 277 deletions.
diff --git a/minigrid/__init__.py b/minigrid/__init__.py
@@ -1136,12 +1136,12 @@ def register_minigrid_envs():
 
     # BabyAI - Language based levels - Level_MixedTrainLocal and Level_MixedTestLocal
     # ----------------------------------------
-    
+
     register(
         id="BabyAI-MixedTrainLocal-v0",
         entry_point="minigrid.envs.babyai:Level_MixedTrainLocal",
     )
-    
+
     register(
         id="BabyAI-MixedTestLocal-v0",
         entry_point="minigrid.envs.babyai:Level_MixedTestLocal",

diff --git a/minigrid/envs/babyai/__init__.py b/minigrid/envs/babyai/__init__.py
@@ -13,6 +13,10 @@
     GoToRedBlueBall,
     GoToSeq,
 )
+from minigrid.envs.babyai.mixed_seq_levels import (
+    Level_MixedTestLocal,
+    Level_MixedTrainLocal,
+)
 from minigrid.envs.babyai.open import (
     Open,
     OpenDoor,
@@ -51,7 +55,3 @@
     UnlockPickup,
     UnlockToUnlock,
 )
-from minigrid.envs.babyai.mixed_seq_levels import (
-    Level_MixedTrainLocal,
-    Level_MixedTestLocal,
-)
diff --git a/minigrid/envs/babyai/core/levelgen.py b/minigrid/envs/babyai/core/levelgen.py
@@ -96,18 +96,9 @@ def add_locked_room(self, color=None):
                 continue
 
             if color is not None:
-                door, _ = self.add_door(
-                    i, j,
-                    door_idx,
-                    color=color,
-                    locked=True
-                )
+                door, _ = self.add_door(i, j, door_idx, color=color, locked=True)
             else:
-                door, _ = self.add_door(
-                    i, j,
-                    door_idx,
-                    locked=True
-                )
+                door, _ = self.add_door(i, j, door_idx, locked=True)
 
             # Done adding locked room
             break