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")),