bundle/v0/scripts/run_paper_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.paper_task import run_gaussian_task


def main() -> int:
    parser = argparse.ArgumentParser(
        description="Run one immutable Loss-aware OT-DRO Gaussian 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,
        help="Optional execution contract. Paper-scale overrides are rejected.",
    )
    args = parser.parse_args()
    requested = load_json(args.execution_config) if args.execution_config else None
    result = run_gaussian_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())