# tar runs/<run-id>/ with model + onnx + eval json + manifest, print sha256
# and the vastai copy command for transfer back to the laptop.

import argparse
from pathlib import Path

from cleanup.pack.ship import pack


def main() -> None:
    parser = argparse.ArgumentParser()
    parser.add_argument("--runs-dir", default="runs")
    parser.add_argument("--dist-dir", default="dist")
    parser.add_argument("--run-id", required=True)
    args = parser.parse_args()

    run_dir = Path(args.runs_dir) / args.run_id
    if not run_dir.exists():
        raise FileNotFoundError(f"no run at {run_dir}")
    pack(run_dir, Path(args.dist_dir), args.run_id)


if __name__ == "__main__":
    main()