register babyai-text

Farama-Foundation · Oct 31, 2024 · 34d3feb · 34d3feb
1 parent a021837
commit 34d3feb
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/minigrid/__init__.py b/minigrid/__init__.py
@@ -1133,3 +1133,16 @@ def register_minigrid_envs():
         id="BabyAI-BossLevelNoUnlock-v0",
         entry_point="minigrid.envs.babyai:BossLevelNoUnlock",
     )
+
+    # 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",
+    )