Fix a tests that uses different dimensions (#223)

marqo-ai · Mar 27, 2024 · a4576d0 · a4576d0
1 parent d3e0cc8
commit a4576d0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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=[
                      {