from __future__ import annotations

import argparse
from pathlib import Path
import sys

PROJECT_ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(PROJECT_ROOT / "src"))

from ml_crypto_lab.core.config import load_yaml
from ml_crypto_lab.train.runner import run_full_experiment


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--config", required=True, help="Path to experiment yaml")
    args = parser.parse_args()
    cfg = load_yaml(args.config)
    result = run_full_experiment(cfg)
    print("=" * 120)
    print("DONE")
    print("run_id:", result["run_id"])
    print("run_dir:", result["run_dir"])
    print("best rows:")
    print(result["summary"].head(20))


if __name__ == "__main__":
    main()