in src/model/get_models.py [0:0]
def get_models() -> List[Model]:
# load from yaml file
with open("src/model/models.yaml", "r") as f:
models = yaml.load(f, Loader=yaml.FullLoader)
models = [Model(**model) for model in models["models"]]
return models