Clean up main.py: remove excluded tasks messaging (all 400 count)
Browse files- 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 |
-
|
| 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,
|
| 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 |
|