| |
| """Bootstrap: cache all 25 public ARC-AGI-3 games locally (one-time, ~3 MB). |
| |
| python ops/fetch_games.py # -> ./environment_files/ |
| python ops/fetch_games.py --dir /data/env # custom cache dir |
| python ops/fetch_games.py --list # live game list from the API |
| |
| After this every eval/collection run uses OperationMode.OFFLINE (zero |
| network, unlimited instances). No Kaggle account or competition rules are |
| required: the arcade API issues an anonymous key automatically. (The |
| kagglehub competition download needs a KGAT token + accepted rules; this |
| toolkit path does not.) |
| """ |
| from __future__ import annotations |
|
|
| import argparse |
| import sys |
| import time |
| from pathlib import Path |
|
|
| ROOT = Path(__file__).resolve().parents[1] |
| sys.path.insert(0, str(ROOT)) |
|
|
| DEFAULT_GAMES = [ |
| "ar25", "bp35", "cd82", "cn04", "dc22", "ft09", "g50t", "ka59", |
| "lf52", "lp85", "ls20", "m0r0", "r11l", "re86", "s5i5", "sb26", |
| "sc25", "sk48", "sp80", "su15", "tn36", "tr87", "tu93", "vc33", "wa30", |
| ] |
|
|
|
|
| def main() -> None: |
| ap = argparse.ArgumentParser(description=__doc__.splitlines()[0]) |
| ap.add_argument("--dir", default=str(Path(__file__).resolve().parents[1] / "environment_files")) |
| ap.add_argument("--list", action="store_true", help="list games and exit") |
| args = ap.parse_args() |
|
|
| import arc_agi |
| from arc_agi import OperationMode |
|
|
| arc = arc_agi.Arcade(operation_mode=OperationMode.NORMAL, |
| environments_dir=args.dir) |
| try: |
| live = [e.game_id.split("-")[0] for e in arc.get_environments()] |
| except Exception as exc: |
| print(f"[warn] API listing failed ({exc}); using static list") |
| live = DEFAULT_GAMES |
| ids = live or DEFAULT_GAMES |
|
|
| if args.list: |
| print("\n".join(ids)) |
| print(f"-- {len(ids)} games available from the arcade API", file=sys.stderr) |
| return |
|
|
| print(f"cache dir: {args.dir}; {len(ids)} games") |
| ok, fail = [], [] |
| for gid in ids: |
| t0 = time.time() |
| try: |
| env = arc.make(gid) |
| if env is None: |
| raise RuntimeError("make() returned None") |
| ok.append(gid) |
| print(f" {gid}: OK ({time.time() - t0:.1f}s)", flush=True) |
| except Exception as exc: |
| fail.append(gid) |
| print(f" {gid}: FAIL {type(exc).__name__}: {exc}", flush=True) |
|
|
| print(f"downloaded={len(ok)} failed={len(fail)}" + (f" {fail}" if fail else "")) |
| sys.exit(1 if fail else 0) |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
