fix max_docs in perform_fusion() (#1589)

castorini · Jul 1, 2021 · e4c5127 · e4c5127
1 parent 52827bb
commit e4c5127
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/python/fusion.py b/src/main/python/fusion.py
@@ -36,7 +36,7 @@ def perform_fusion(method: FusionMethod, runs: List[TrecRun], output_path: str,
     print('Performing fusion ->', method)
 
     if method == FusionMethod.RRF:
-        fused_run = fusion.reciprocal_rank_fusion(runs, max_docs=10000)
+        fused_run = fusion.reciprocal_rank_fusion(runs, max_docs=max_docs)
         fused_run.print_subset(output_path, topics=fused_run.topics())
     elif method == FusionMethod.COMBO_SUM:
         with open(output_path, 'w+') as f: