mcontriever fix test (castorini#1550)

yilinjz · Jun 12, 2023 · 2071545 · 2071545
1 parent faa4003
commit 2071545
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_prebuilt_index.py b/tests/test_prebuilt_index.py
@@ -132,8 +132,8 @@ def test_faiss_miracl(self):
                 for url in FAISS_INDEX_INFO[key]['urls']:
                     urls.append(url)
 
-        # 18 pFT MS MARCO, 18 pFT MS MARCO all, 16 pFT MS MARCO + per lang (no de, yo)
-        self.assertEqual(cnt, 52)
+        # 18 pFT MS MARCO, 18 pFT MS MARCO all, 16 pFT MS MARCO + per lang (no de, yo), 18 mContriever pFT MS MARCO
+        self.assertEqual(cnt, 70)
         self._test_urls(urls)
 
     def test_faiss_msmarco(self):