Spaces:
Sleeping
Sleeping
| """Quick smoke test: checks all 3 required items for repair_env_v2.""" | |
| import sys, os | |
| sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) | |
| root = os.path.dirname(os.path.abspath(__file__)) | |
| print("=" * 50) | |
| print(" repair_env_v2 — Pre-flight Check") | |
| print("=" * 50) | |
| # --- 1. Dockerfile --- | |
| df = os.path.join(root, "Dockerfile") | |
| ok1 = os.path.isfile(df) | |
| print(f"[{'OK' if ok1 else 'MISSING'}] Dockerfile") | |
| # --- 2. inference.py --- | |
| inf = os.path.join(root, "inference.py") | |
| ok2 = os.path.isfile(inf) | |
| print(f"[{'OK' if ok2 else 'MISSING'}] inference.py") | |
| # --- 3. openenv.yaml --- | |
| yaml_f = os.path.join(root, "openenv.yaml") | |
| ok3 = os.path.isfile(yaml_f) | |
| print(f"[{'OK' if ok3 else 'MISSING'}] openenv.yaml") | |
| # --- 4. inference.py actually runs --- | |
| print("\n--- Running inference on all 4 tasks (verbose=False) ---") | |
| try: | |
| import inference as inf_mod | |
| from tasks_v2 import TASKS | |
| results = [] | |
| for tid in TASKS: | |
| r = inf_mod.run_inference(task_id=tid, verbose=False) | |
| results.append(r) | |
| print(f"\n{'Task':<12} {'Steps':>5} {'Optimal':>7} {'Reward':>7} {'Score':>6} {'Success':>7} {'Health':>6}") | |
| print("-" * 60) | |
| for r in results: | |
| opt = TASKS[r["task_id"]]["optimal_steps"] | |
| print(f"{r['task_id']:<12} {r['steps_taken']:>5} {opt:>7} {r['total_reward']:>7.3f} {r['score']:>6.3f} {str(r['success']):>7} {r['final_health_ratio']:>6.2%}") | |
| print("\n[OK] inference.py runs cleanly on all tasks") | |
| except Exception as e: | |
| print(f"[FAIL] inference.py error: {e}") | |
| import traceback; traceback.print_exc() | |
| print("\n" + "=" * 50) | |
| all_ok = ok1 and ok2 and ok3 | |
| print("RESULT:", "ALL CHECKS PASSED" if all_ok else "SOME CHECKS FAILED") | |
| print("=" * 50) | |