rogermt commited on
Commit
5598fb7
·
verified ·
1 Parent(s): f6b5eb9

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