feat(overcooked): Include new layout

FLAIROx · Jun 25, 2024 · 82301b5 · 82301b5
1 parent 0eabb6a
commit 82301b5
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/jaxmarl/environments/overcooked_v2/layouts.py b/jaxmarl/environments/overcooked_v2/layouts.py
@@ -89,6 +89,12 @@
 WWWWWWWWW
 """
 
+long_room = """
+WWWWWWWWWWWWWWW
+B            AP
+0             X
+WWWWWWWWWWWWWWW
+"""
 
 @dataclass
 class Layout:
@@ -215,4 +221,5 @@ def layout_grid_to_dict(grid, recipe=None):
     "counter_circuit": layout_grid_to_dict(counter_circuit_grid, recipe=[0, 0, 0]),
     "two_rooms": layout_grid_to_dict(two_rooms),
     "two_rooms_simple": layout_grid_to_dict(two_rooms_simple),
+    "long_room": layout_grid_to_dict(long_room, recipe=[0, 0, 0]),
 }