code/repro_scripts/hf_job_claim3.py

# /// script
# requires-python = ">=3.11"
# dependencies = [
#   "torch",
#   "numpy",
#   "scipy",
#   "matplotlib",
#   "tqdm",
#   "pyyaml",
# ]
# ///
"""HF Jobs wrapper: Claim 3 CZ adaptation under 10x noise."""
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" / "fig_5_two_qubit_cz"))
os.chdir(CODE)
# Point nested script outputs at bucket
os.environ.setdefault("CLAIM3_ITERS", "1000")
runpy.run_path(str(CODE / "repro_scripts" / "verify_claim3_gpu.py"), run_name="__main__")