linting

src/workrb/rankings.py

@@ -140,9 +140,7 @@ def materialize_prediction_matrix(
     matrix = np.zeros((num_queries, num_targets), dtype=np.float32)
     for q_idx, row in scores.items():
         if not 0 <= q_idx < num_queries:
-            raise IndexError(
-                f"query_index {q_idx} out of bounds for num_queries={num_queries}"
-            )
+            raise IndexError(f"query_index {q_idx} out of bounds for num_queries={num_queries}")
         for t_idx, score in row.items():
             if not 0 <= t_idx < num_targets:
                 raise IndexError(

src/workrb/tasks/abstract/ranking_base.py

@@ -476,4 +476,3 @@ def compute_metrics_from_prediction_matrix(
             pos_label_relevance=dataset.target_relevance,
             binary_relevance_threshold=self.binary_relevance_threshold,
         )
-

tests/test_save_rankings_artifacts.py

@@ -338,9 +338,7 @@ def test_save_rankings_rejects_non_finite_scores():
 
     from workrb.config import BenchmarkConfig
 
-    dataset = TinyRankingTask(
-        split=DatasetSplit.TEST, languages=[Language.EN]
-    ).datasets["en"]
+    dataset = TinyRankingTask(split=DatasetSplit.TEST, languages=[Language.EN]).datasets["en"]
     config = BenchmarkConfig(
         model_name="tiny",
         output_folder=str(_fresh_dir("rankings_artifact_non_finite_write")),