rogermt commited on
Commit
909a0d3
·
verified ·
1 Parent(s): 22b8d94

Clean up main.py: remove excluded tasks messaging (all 400 count)

Browse files
Files changed (1) hide show
  1. neurogolf_solver/main.py +2 -4
neurogolf_solver/main.py CHANGED
@@ -56,11 +56,9 @@ def main():
56
 
57
  total_arcgen = sum(len(t['data'].get('arc-gen', [])) for t in tasks.values())
58
  print(f"Loaded {len(tasks)} tasks ({total_arcgen} ARC-GEN examples)")
59
- print(f"Excluded tasks: {sorted(EXCLUDED_TASKS)}")
60
 
61
  task_nums = [int(t) for t in args.tasks.split(',')] if args.tasks else sorted(tasks.keys())
62
- active_tasks = [t for t in task_nums if t not in EXCLUDED_TASKS]
63
- print(f"Solving {len(active_tasks)} active tasks (skipping {len(task_nums) - len(active_tasks)} excluded)")
64
  print(f"Conv budget: {args.conv_budget}s per task")
65
  print("=" * 70)
66
 
@@ -80,7 +78,7 @@ def main():
80
 
81
  elapsed = time.time() - t0
82
 
83
- submission_info = generate_submission(args.output_dir, results, costs_dict, active_tasks)
84
  print_summary(results, submission_info, elapsed)
85
 
86
 
 
56
 
57
  total_arcgen = sum(len(t['data'].get('arc-gen', [])) for t in tasks.values())
58
  print(f"Loaded {len(tasks)} tasks ({total_arcgen} ARC-GEN examples)")
 
59
 
60
  task_nums = [int(t) for t in args.tasks.split(',')] if args.tasks else sorted(tasks.keys())
61
+ print(f"Solving {len(task_nums)} tasks")
 
62
  print(f"Conv budget: {args.conv_budget}s per task")
63
  print("=" * 70)
64
 
 
78
 
79
  elapsed = time.time() - t0
80
 
81
+ submission_info = generate_submission(args.output_dir, results, costs_dict, task_nums)
82
  print_summary(results, submission_info, elapsed)
83
 
84