ruff lint

Signed-off-by: Adam Gutglick <adam@spiraldb.com>

Author: AdamGS

bench-orchestrator/bench_orchestrator/cli.py

@@ -97,7 +97,9 @@ def backends_for_engines(engines: list[Engine]) -> list[ExecutionBackend]:
         Engine.DUCKDB: Format.PARQUET,
         Engine.LANCE: Format.LANCE,
     }
-    return list(group_targets_by_backend(BenchmarkTarget(engine=engine, format=seed_formats[engine]) for engine in engines))
+    return list(
+        group_targets_by_backend(BenchmarkTarget(engine=engine, format=seed_formats[engine]) for engine in engines)
+    )
 
 
 @contextmanager
@@ -288,10 +290,12 @@ def run(
                         samply=samply,
                         sample_rate=sample_rate,
                         tracing=tracing,
-                        on_result=lambda line, store_writer=ctx.write_raw_json, compatibility=compatibility_file: write_result_line(
-                            line,
-                            store_writer,
-                            compatibility,
+                        on_result=lambda line, store_writer=ctx.write_raw_json, compatibility=compatibility_file: (
+                            write_result_line(
+                                line,
+                                store_writer,
+                                compatibility,
+                            )
                         ),
                     )
                     console.print(f"[green]{backend.value}: {len(results)} results[/green]")
@@ -540,8 +544,7 @@ def show(
     console.print(f"  Formats: {', '.join(metadata.formats)}")
     if metadata.targets:
         console.print(
-            "  Targets: "
-            + ", ".join(f"{target['engine']}:{target['format']}" for target in metadata.targets)
+            "  Targets: " + ", ".join(f"{target['engine']}:{target['format']}" for target in metadata.targets)
         )
     console.print(f"  Iterations: {metadata.iterations}")
     console.print(f"  Git commit: {metadata.git_commit[:8]}")

bench-orchestrator/bench_orchestrator/config.py

@@ -180,7 +180,9 @@ def parse_formats_json(value: str) -> list[Format]:
     return _unique_preserve_order(formats)
 
 
-def resolve_axis_targets(engines: Iterable[Engine], formats: Iterable[Format]) -> tuple[list[BenchmarkTarget], list[str]]:
+def resolve_axis_targets(
+    engines: Iterable[Engine], formats: Iterable[Format]
+) -> tuple[list[BenchmarkTarget], list[str]]:
     """Expand engine/format axes into supported explicit targets."""
     warnings: list[str] = []
     targets: list[BenchmarkTarget] = []

bench-orchestrator/tests/test_cli.py

@@ -2,13 +2,11 @@
 # SPDX-FileCopyrightText: Copyright the Vortex contributors
 
 import json
-from pathlib import Path
-
-from typer.testing import CliRunner
 
 from bench_orchestrator import cli as cli_module
 from bench_orchestrator.runner.executor import BenchmarkExecutor
 from bench_orchestrator.storage.store import ResultStore
+from typer.testing import CliRunner
 
 runner = CliRunner()
 

bench-orchestrator/tests/test_config.py

@@ -15,9 +15,7 @@
 
 
 def test_parse_targets_json_normalizes_and_dedupes_lance_targets() -> None:
-    targets = parse_targets_json(
-        '[{"engine":"lance","format":"lance"},{"engine":"datafusion","format":"lance"}]'
-    )
+    targets = parse_targets_json('[{"engine":"lance","format":"lance"},{"engine":"datafusion","format":"lance"}]')
 
     assert targets == [BenchmarkTarget(engine=Engine.DATAFUSION, format=Format.LANCE)]
 
@@ -65,6 +63,4 @@ def test_group_targets_by_backend_routes_lance_to_lance_binary() -> None:
         ExecutionBackend.LANCE,
         ExecutionBackend.DUCKDB,
     ]
-    assert groups[ExecutionBackend.LANCE] == [
-        BenchmarkTarget(engine=Engine.DATAFUSION, format=Format.LANCE)
-    ]
+    assert groups[ExecutionBackend.LANCE] == [BenchmarkTarget(engine=Engine.DATAFUSION, format=Format.LANCE)]