|
27 | 27 |
|
28 | 28 | from swe_forge.export.jsonl import export_jsonl |
29 | 29 | from swe_forge.swe.github_api import GitHubClient |
30 | | -from swe_forge.swe.harness import HarnessConfig, HarnessRunner, HarnessStatus |
| 30 | +from swe_forge.swe.harness import HarnessConfig, HarnessStatus |
31 | 31 | from swe_forge.swe.models import SweTask |
32 | 32 | from swe_forge.swe.pipeline import SwePipeline, SwePipelineConfig |
33 | 33 |
|
@@ -139,9 +139,7 @@ def benchmark( |
139 | 139 | ) |
140 | 140 |
|
141 | 141 | if difficulty and not validate_difficulty(difficulty): |
142 | | - console.print( |
143 | | - f"[red]Error: Difficulty must be one of: easy, medium, hard[/red]" |
144 | | - ) |
| 142 | + console.print("[red]Error: Difficulty must be one of: easy, medium, hard[/red]") |
145 | 143 | raise typer.Exit(code=1) |
146 | 144 |
|
147 | 145 | if not model.strip(): |
@@ -276,10 +274,7 @@ async def _mine_tasks( |
276 | 274 | task_id = progress.add_task("Mining tasks...", total=num_tasks) |
277 | 275 |
|
278 | 276 | async with SwePipeline(gh_client, config=config) as pipeline: |
279 | | - from swe_forge.swe.pipeline import ( |
280 | | - SwePipelineEvent, |
281 | | - SwePipelineEventType, |
282 | | - ) |
| 277 | + from swe_forge.swe.pipeline import SwePipelineEventType |
283 | 278 |
|
284 | 279 | async for event in pipeline.run_with_progress(): |
285 | 280 | if event.event_type == SwePipelineEventType.TASK_EXTRACTED: |
@@ -468,7 +463,6 @@ def _print_summary(results: list[dict]) -> None: |
468 | 463 |
|
469 | 464 | for status, count in sorted(status_counts.items()): |
470 | 465 | pct = (count / total_count * 100) if total_count > 0 else 0 |
471 | | - style = "green" if status == HarnessStatus.RESOLVED.value else "red" |
472 | 466 | table.add_row(status, str(count), f"{pct:.1f}%") |
473 | 467 |
|
474 | 468 | console.print(table) |
|
0 commit comments