Fix: delete leftover .onnx on unsolved tasks, --strict_score default False
Browse files
neurogolf_solver/solvers/solver_registry.py
CHANGED
|
@@ -75,10 +75,21 @@ def _accept_model(path, td, providers):
|
|
| 75 |
return True
|
| 76 |
|
| 77 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 78 |
def solve_task(tn, td, outdir, providers, conv_budget=30.0, excluded_tasks=None):
|
| 79 |
"""Solve a single ARC-AGI task.
|
| 80 |
|
| 81 |
Returns: (ok, solver_name, file_size, elapsed, model_path)
|
|
|
|
| 82 |
"""
|
| 83 |
if excluded_tasks is None:
|
| 84 |
excluded_tasks = EXCLUDED_TASKS
|
|
@@ -139,4 +150,6 @@ def solve_task(tn, td, outdir, providers, conv_budget=30.0, excluded_tasks=None)
|
|
| 139 |
sname, model = result
|
| 140 |
return True, sname, os.path.getsize(path), time.time() - t_start, path
|
| 141 |
|
|
|
|
|
|
|
| 142 |
return False, None, None, time.time() - t_start, path
|
|
|
|
| 75 |
return True
|
| 76 |
|
| 77 |
|
| 78 |
+
def _cleanup_failed(path):
|
| 79 |
+
"""Delete leftover .onnx file from failed solve attempts.
|
| 80 |
+
Prevents bad files from ending up in submission zip."""
|
| 81 |
+
try:
|
| 82 |
+
if os.path.exists(path):
|
| 83 |
+
os.remove(path)
|
| 84 |
+
except OSError:
|
| 85 |
+
pass
|
| 86 |
+
|
| 87 |
+
|
| 88 |
def solve_task(tn, td, outdir, providers, conv_budget=30.0, excluded_tasks=None):
|
| 89 |
"""Solve a single ARC-AGI task.
|
| 90 |
|
| 91 |
Returns: (ok, solver_name, file_size, elapsed, model_path)
|
| 92 |
+
If unsolved, deletes any leftover .onnx file.
|
| 93 |
"""
|
| 94 |
if excluded_tasks is None:
|
| 95 |
excluded_tasks = EXCLUDED_TASKS
|
|
|
|
| 150 |
sname, model = result
|
| 151 |
return True, sname, os.path.getsize(path), time.time() - t_start, path
|
| 152 |
|
| 153 |
+
# All solvers failed — delete leftover .onnx file
|
| 154 |
+
_cleanup_failed(path)
|
| 155 |
return False, None, None, time.time() - t_start, path
|