bundle/v0/scripts/run_task.py

from __future__ import annotations

import argparse
import json
from pathlib import Path

from loss_aware_dro_repro.core import load_json
from loss_aware_dro_repro.task_dispatch import run_task


def main() -> int:
    parser = argparse.ArgumentParser(
        description="Run one immutable Loss-aware OT-DRO matrix task."
    )
    parser.add_argument("--task-id", required=True)
    parser.add_argument("--output-dir", type=Path, required=True)
    parser.add_argument("--execution-config", type=Path)
    args = parser.parse_args()
    requested = load_json(args.execution_config) if args.execution_config else None
    result = run_task(
        args.task_id,
        args.output_dir,
        requested_execution_config=requested,
    )
    print(json.dumps(result, indent=2, sort_keys=True))
    return 0


if __name__ == "__main__":
    raise SystemExit(main())