model.py

from __future__ import annotations
from typing import List, Dict, Tuple, Union
import sys
from pathlib import Path
from pprint import pprint

hytraits_path = (Path(__file__).parent.parent/'hytraits').resolve()
if str(hytraits_path) not in sys.path:
    sys.path.append(str(hytraits_path))
import hytraits as H 

from utils import get_cli_args
from project import (make_compatible_csvs,
                     get_model_names,
                     train_model,
                     deploy_model)


if __name__ == '__main__':
    ARGS = get_cli_args()

    make_compatible_csvs()
    
    model_names = get_model_names(pattern=ARGS['pattern'])
    for model_name in model_names:
        train_model(model_name=model_name, args=ARGS)
        deploy_model(model_name=model_name, args=ARGS)