from pathlib import Path
import sys

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

from model.atmosphericda_dmodel import load_yaml, run_inference


if __name__ == "__main__":
    config = load_yaml(ROOT / "conf/config.yaml")
    metrics = run_inference(
        str(ROOT / config["paths"]["data"]),
        str(ROOT / config["paths"]["checkpoint"]),
        str(ROOT / config["paths"]["predictions"]), config["project"]["format_version"],
    )
    print(f"saved predictions: {metrics['hybrid_prediction'].shape}")