From 9db25847829a656d1c9eacb267bf745f7522dd14 Mon Sep 17 00:00:00 2001
From: Jasper Xian <41269031+jasper-xian@users.noreply.github.com>
Date: Wed, 24 Apr 2024 23:04:51 -0400
Subject: [PATCH] Add unit test for BEIR Cohere embed-english-v3.0 indexes
 (#1871)

---
 tests/test_prebuilt_index.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_prebuilt_index.py b/tests/test_prebuilt_index.py
index 2c84038e7..0a14f7c31 100644
--- 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):