From a4576d0e909d436c8bfd8157b0835682d45180c3 Mon Sep 17 00:00:00 2001
From: Li Wan <49334982+wanliAlex@users.noreply.github.com>
Date: Wed, 27 Mar 2024 11:37:13 +1100
Subject: [PATCH] Fix a tests that uses different dimensions (#223)

---
 tests/v2_tests/test_add_documents.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/v2_tests/test_add_documents.py b/tests/v2_tests/test_add_documents.py
index e2874834..51ad1977 100644
--- a/tests/v2_tests/test_add_documents.py
+++ b/tests/v2_tests/test_add_documents.py
@@ -510,13 +510,14 @@ def test_custom_vector_doc(self):
          3. tensor search
          4. get document
          """
-
-         DEFAULT_DIMENSIONS = 768
          for cloud_test_index_to_use, open_source_test_index_name in self.test_cases:
              test_index_name = self.get_test_index_name(
                  cloud_test_index_to_use=cloud_test_index_to_use,
                  open_source_test_index_name=open_source_test_index_name
              )
+
+             DEFAULT_DIMENSIONS = 512 if "img" in test_index_name else 768
+
              add_docs_res = self.client.index(index_name=test_index_name).add_documents(
                  documents=[
                      {