serviceadvisor / car_core /specs_model_space.py
viswanani's picture
Upload 16 files
94332c9 verified
raw
history blame contribute delete
296 Bytes
import yaml
def build_label_space(models_yaml="configs/models.yaml"):
cfg = yaml.safe_load(open(models_yaml, "r", encoding="utf-8"))
space = []
for make, models in cfg.get("makes_models", {}).items():
for m in models:
space.append(f"{make} {m}")
return space