Spaces:
Sleeping
Sleeping
| import argparse | |
| import subprocess | |
| import sys | |
| def run_command(cmd: list[str]) -> None: | |
| print("Running:", " ".join(cmd)) | |
| result = subprocess.run(cmd, check=False) | |
| if result.returncode != 0: | |
| raise RuntimeError(f"Command failed with exit code {result.returncode}: {' '.join(cmd)}") | |
| def main(): | |
| parser = argparse.ArgumentParser(description="Run end-to-end local pipeline") | |
| parser.add_argument("--base-url", default="http://localhost:8000") | |
| parser.add_argument("--episodes", type=int, default=60) | |
| parser.add_argument("--eval-episodes", type=int, default=20) | |
| args = parser.parse_args() | |
| py = sys.executable | |
| run_command([py, "scripts/smoke_test.py", "--base-url", args.base_url]) | |
| run_command( | |
| [ | |
| py, | |
| "training/run_training.py", | |
| "--base-url", | |
| args.base_url, | |
| "--episodes", | |
| str(args.episodes), | |
| "--eval-episodes", | |
| str(args.eval_episodes), | |
| ] | |
| ) | |
| run_command([py, "benchmarks/run_baselines.py", "--base-url", args.base_url, "--episodes", "30"]) | |
| run_command([py, "scripts/plot_metrics.py", "--input", "artifacts/training_rollout.json"]) | |
| print("Pipeline complete. Artifacts are under artifacts/ and artifacts/plots/.") | |
| if __name__ == "__main__": | |
| main() | |