From c4d9885a1ed10b800b2ecdb1f2a7e0a6d6a1bbc4 Mon Sep 17 00:00:00 2001 From: "warre.veys" Date: Tue, 26 May 2026 17:00:38 +0200 Subject: [PATCH] linting --- src/workrb/rankings.py | 4 +--- src/workrb/tasks/abstract/ranking_base.py | 1 - tests/test_save_rankings_artifacts.py | 4 +--- 3 files changed, 2 insertions(+), 7 deletions(-) diff --git a/src/workrb/rankings.py b/src/workrb/rankings.py index a61c165..ac3f501 100644 --- a/src/workrb/rankings.py +++ b/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( diff --git a/src/workrb/tasks/abstract/ranking_base.py b/src/workrb/tasks/abstract/ranking_base.py index 00a5842..581cce7 100644 --- a/src/workrb/tasks/abstract/ranking_base.py +++ b/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, ) - diff --git a/tests/test_save_rankings_artifacts.py b/tests/test_save_rankings_artifacts.py index b17bdc2..e6d6037 100644 --- a/tests/test_save_rankings_artifacts.py +++ b/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")),