Merge pull request stanford-futuredata#2 from seanmacavaney/patch-2

fixed assert in slow_rerank
Yaasha · Mar 26, 2021 · 4367fae · 4367fae
2 parents cc2424f + e7e44b6
commit 4367fae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/colbert/evaluation/slow.py b/colbert/evaluation/slow.py
@@ -16,6 +16,6 @@ def slow_rerank(args, query, pids, passages):
     ranked_pids = [pids[position] for position in ranked]
     ranked_passages = [passages[position] for position in ranked]
 
-    assert len(ranked_pids) == len(set(ranked_pids))
+    assert len(ranked_pids) == len(ranked_scores)
 
     return list(zip(ranked_scores, ranked_pids, ranked_passages))