code/repro_scripts/hf_job_claim5.py

# /// script
# requires-python = ">=3.11"
# dependencies = [
#   "torch",
#   "numpy",
#   "scipy",
#   "matplotlib",
#   "tqdm",
#   "pyyaml",
# ]
# ///
"""HF Jobs wrapper: Claim 5 / Corollary 4.10 variance sweep."""
from __future__ import annotations

import os
import runpy
import sys
from pathlib import Path

CODE = Path(os.environ.get("CODE_ROOT", "/data/code"))
OUT = Path(os.environ.get("OUT_ROOT", "/data/outputs"))
OUT.mkdir(parents=True, exist_ok=True)
sys.path.insert(0, str(CODE))
sys.path.insert(0, str(CODE / "experiments"))
os.chdir(CODE)
os.environ.setdefault("CLAIM5_ITERS", "300")
os.environ.setdefault("CLAIM5_BATCH", "8")
runpy.run_path(str(CODE / "repro_scripts" / "verify_claim5_gpu.py"), run_name="__main__")