diff --git a/overcooked_simulator/gui_2d_vis/overcooked_gui.py b/overcooked_simulator/gui_2d_vis/overcooked_gui.py index 2ab2e9001131b2587ae743fe3256a758f5509c22..26fa56ec744d3993c5c19f20ed9499ea2358ec9f 100644 --- a/overcooked_simulator/gui_2d_vis/overcooked_gui.py +++ b/overcooked_simulator/gui_2d_vis/overcooked_gui.py @@ -1,10 +1,12 @@ import argparse import dataclasses +import glob import json import logging import random import sys from enum import Enum +from pathlib import Path from subprocess import Popen import numpy as np @@ -357,16 +359,9 @@ class PyGameGUI: self.layout_file_paths = { str(p.name): p - for p in (ROOT_DIR / "game_content" / "layouts").glob("*.layout") + for p in [Path(f) for f in sorted((ROOT_DIR / "game_content" / "layouts").rglob("*.layout"))] } - self.layout_file_paths.update( - { - str(p.name): p - for p in (ROOT_DIR / "game_content" / "layouts" / "overcooked-1").glob( - "*.layout" - ) - } - ) + assert len(self.layout_file_paths) != 0, "No layout files." dropdown_width, dropdown_height = 200, 40 self.layout_selection = pygame_gui.elements.UIDropDownMenu(