Update Down/run_all.py

Down/run_all.py

@@ -1,76 +1,45 @@
-"""Run both DOWN and AntiDown tabular experiments.
-
-Usage:
-  python run_all.py --episodes 400
-
-This will:
-- run down tabular paradox experiment
-- run AntiDown corrupted valley tabular experiment
-- generate plots for each into their graphics/ folders
-"""
-
 from __future__ import annotations
-
-# Headless-safe plotting (Hugging Face / Linux without DISPLAY)
 import matplotlib
 matplotlib.use("Agg")
 
 import argparse
 import os
 import subprocess
-from pathlib import Path
 import sys
+from pathlib import Path
 
+ROOT = Path(__file__).resolve().parent
+DOWN_DIR = ROOT / "down"
+ANTIDOWN_DIR = ROOT / "AntiDown"
+OUT_GRAPHICS = ROOT / "graphics"
 
-def _run(cmd: list[str], cwd: Path, env: dict[str, str] | None = None) -> None:
+def _run(cmd, cwd: Path):
     print(f"[run] ({cwd}) $ {' '.join(cmd)}")
-    merged = os.environ.copy()
-    if env:
-        merged.update(env)
-    subprocess.run(cmd, cwd=str(cwd), check=True, env=merged)
-
-
-def main() -> None:
-    ap = argparse.ArgumentParser()
-    ap.add_argument("--episodes", type=int, default=400, help="Episodes per phase (or main loop) where applicable")
-    args = ap.parse_args()
+    subprocess.run(cmd, cwd=str(cwd), check=True)
 
-    root = Path(__file__).resolve().parent
-    results_dir = root / "results"
-    graphics_dir = root / "graphics"
-    results_dir.mkdir(parents=True, exist_ok=True)
-    graphics_dir.mkdir(parents=True, exist_ok=True)
 
-    # Common env for all subprocesses (plots must be headless-safe)
-    common_env = {
-        "TWOQUARKS_RESULTS_DIR": results_dir.as_posix(),
-        "TWOQUARKS_GRAPHICS_DIR": graphics_dir.as_posix(),
-        "MPLBACKEND": "Agg",  # <-- critical for HF headless plotting
-    }
-
-    down_dir = root / "down"
-    AntiDown_dir = root / "AntiDown"
+def main():
+    py = sys.executable
+    OUT_GRAPHICS.mkdir(parents=True, exist_ok=True)
 
     # DOWN
-    down_script = down_dir / "exp" / "run_paradox_tabular.py"
-    _run([sys.executable, str(down_script)], cwd=down_dir, env=common_env)
-
-    plot_script = down_dir / "exp" / "plot_paradox_results.py"
-    _run([sys.executable, str(plot_script)], cwd=down_dir, env=common_env)
+    _run([py, str(DOWN_DIR / "exp" / "run_paradox_tabular.py")], cwd=DOWN_DIR)
+    _run([py, str(DOWN_DIR / "exp" / "plot_paradox_results.py"),
+          "--csv", str(DOWN_DIR / "results" / "paradox_tabular_results.csv"),
+          "--out", str(OUT_GRAPHICS),
+          "--prefix", "down"], cwd=DOWN_DIR)
 
     # AntiDown
-    AntiDown_script = AntiDown_dir / "exp" / "run_corrupted_valley_tabular.py"
-    _run([sys.executable, str(AntiDown_script)], cwd=AntiDown_dir, env=common_env)
+    _run([py, str(ANTIDOWN_DIR / "exp" / "run_corrupted_valley_tabular.py")],  cwd=ANTIDOWN_DIR)
+    _run([py, str(ANTIDOWN_DIR / "exp" / "plot_corrupted_valley_results.py"),
+          "--csv", str(ANTIDOWN_DIR / "results" / "paradox_tabular_results.csv"),
+          "--out", str(OUT_GRAPHICS),
+          "--prefix", "AntiDown"], cwd=ANTIDOWN_DIR)    
 
-    AntiDown_plot = AntiDown_dir / "exp" / "plot_corrupted_valley_results.py"
-    _run([sys.executable, str(AntiDown_plot)], cwd=AntiDown_dir, env=common_env)
+    _run([py, str(ROOT / "plot_dual_down_antidown.py")], cwd=ROOT)
 
-    # Dual overlay plot (Down vs AntiDown)
-    dual_plot = root / "plot_dual_down_antidown.py"
-    if dual_plot.exists():
-        _run([sys.executable, str(dual_plot)], cwd=root, env=common_env)
+    print(f"[ok] Finished. All plots saved: {OUT_GRAPHICS}")
 
-    print("\nDone. Results and plots saved into root/results and root/graphics.")
 
 
 if __name__ == "__main__":