File size: 296 Bytes
94332c9
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
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