Add unit test for BEIR Cohere embed-english-v3.0 indexes (#1871)

castorini · Apr 25, 2024 · 9db2584 · 9db2584
1 parent e68fac6
commit 9db2584
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_prebuilt_index.py b/tests/test_prebuilt_index.py
@@ -119,8 +119,8 @@ def test_faiss_beir(self):
                 for url in FAISS_INDEX_INFO[key]['urls']:
                     urls.append(url)
 
-        # each 29: contriever, contriever-msmarco, bge
-        self.assertEqual(cnt, 87)
+        # each 29: contriever, contriever-msmarco, bge, cohere-embed-english-v3.0
+        self.assertEqual(cnt, 116)
         self._test_urls(urls)
 
     def test_faiss_mrtydi(self):