Update Down/down/exp/run_paradox_tabular.py

Down/down/exp/run_paradox_tabular.py

@@ -1,16 +1,16 @@
-
 # --- twoquarks bootstrap (path-stable imports) ---
 import sys
 from pathlib import Path
+
 _ROOT = Path(__file__).resolve().parents[1]  # project root (sibling of exp/)
 if str(_ROOT) not in sys.path:
     sys.path.insert(0, str(_ROOT))
 # -------------------------------------------------
 
+import argparse
 import csv
 import os
 from pathlib import Path
-import sys
 
 import numpy as np
 
@@ -35,12 +35,10 @@ def run_experiment(
     Run HF-Levo and Levo Paradox tabular agents and log results to CSV.
 
     The CSV schema is:
-
         global_episode,phase,episode,env_seed,agent,episode_reward,failure_mode,rho_state
     """
     out_path = Path(out_csv)
     out_path.parent.mkdir(parents=True, exist_ok=True)
-    out_path.parent.mkdir(parents=True, exist_ok=True)
 
     rng = np.random.default_rng(seed)
 
@@ -74,12 +72,12 @@ def run_experiment(
         writer.writeheader()
 
         for phase in (1, 2, 3):
-            for local_ep in range(episodes_per_phase):
+            for local_ep in range(int(episodes_per_phase)):
                 env_seed = int(rng.integers(0, 2**31 - 1))
 
                 for agent, name in zip(agents, agent_names, strict=True):
                     env = EpistemicValleyEnv(phase=phase, seed=env_seed)
-                    s_idx, info = env.reset()
+                    s_idx, _info = env.reset()
                     a = agent.select_action(s_idx)
                     s_next_idx, reward, done, step_info = env.step(a)
 
@@ -116,9 +114,26 @@ def run_experiment(
                 global_ep += 1
 
 
+def _build_argparser() -> argparse.ArgumentParser:
+    p = argparse.ArgumentParser(description="Run DOWN paradox tabular experiment and save CSV.")
+    p.add_argument("--episodes", type=int, default=400, help="Episodes per phase (phases: 1,2,3).")
+    p.add_argument("--seed", type=int, default=2025, help="RNG seed.")
+    p.add_argument(
+        "--out",
+        type=str,
+        default="",
+        help="Optional output CSV path. If omitted, uses TWOQUARKS_RESULTS_DIR or down/results.",
+    )
+    return p
+
+
 if __name__ == "__main__":
+    args = _build_argparser().parse_args()
+
     # Allow a caller (e.g., run_all.py) to route outputs to a shared folder.
-    _results_dir = Path(os.environ.get("TWOQUARKS_RESULTS_DIR", (_QUARK_DIR / "results").as_posix()))
-    _results_dir.mkdir(parents=True, exist_ok=True)
-    _out_csv = _results_dir / "down_paradox_tabular_results.csv"
-    run_experiment(_out_csv.as_posix(), episodes_per_phase=400)
+    results_dir = Path(os.environ.get("TWOQUARKS_RESULTS_DIR", (_QUARK_DIR / "results").as_posix()))
+    results_dir.mkdir(parents=True, exist_ok=True)
+
+    out_csv = Path(args.out) if args.out.strip() else (results_dir / "down_paradox_tabular_results.csv")
+
+    run_experiment(out_csv.as_posix(), episodes_per_phase=int(args.episodes), seed=int(args.seed))