he_detail.py 799 B

12345678910111213141516
  1. """Per-task HumanEval results for several saved runs (inside a --network none container)."""
  2. import glob, os, subprocess, sys
  3. runs = sys.argv[1:]
  4. res = {}
  5. for r in runs:
  6. for f in sorted(glob.glob(f"/q/{r}/he/*.py")):
  7. t = os.path.basename(f)[:-3]
  8. try:
  9. ok = subprocess.run([sys.executable, f], capture_output=True, timeout=15).returncode == 0
  10. except subprocess.TimeoutExpired:
  11. ok = False
  12. res.setdefault(t, {})[r] = ok
  13. print("runs:", " ".join(f"{r}={sum(v.get(r, False) for v in res.values())}" for r in runs))
  14. for t, v in sorted(res.items()):
  15. if len(set(v.values())) > 1 or not any(v.values()):
  16. print(f"{t:18} " + " ".join("pass" if v.get(r) else "FAIL" for r in runs) + (" (fails everywhere)" if not any(v.values()) else ""))