from pathlib import Path
import sys

ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(ROOT))

from model.atmosphericda_dmodel import QGConfig, load_yaml, train_model


if __name__ == "__main__":
    config = load_yaml(ROOT / "conf/config.yaml")
    config["training"]["format_version"] = config["project"]["format_version"]
    history = train_model(
        str(ROOT / config["paths"]["data"]),
        str(ROOT / config["paths"]["checkpoint"]),
        str(ROOT / config["paths"]["training_metrics"]),
        QGConfig(**config["qg"]), config["model"], config["training"],
    )
    print(f"loss={history['loss']}")