test image inference only

marqo-ai · Oct 15, 2024 · 4bdd2fb · 4bdd2fb
1 parent 14edba0
commit 4bdd2fb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/perf_tests/batch_add_doc_with_image.py b/perf_tests/batch_add_doc_with_image.py
@@ -44,7 +44,7 @@ def add_docs(self):
                     "image": "https://raw.githubusercontent.com/marqo-ai/marqo/mainline/examples/ImageSearchGuide/data/image4.jpg"
                 },
             ] for i in range(int(32 / 4))] for doc in docs],
-            tensor_fields=["title"],
+            tensor_fields=["image"],
             # mappings={"my_multi_modal_field": {"type": "multimodal_combination",
             #                                    "weights": {"title": 0.5, "image": 0.8}}}
         )

diff --git a/perf_tests/search_with_image.py b/perf_tests/search_with_image.py
@@ -19,7 +19,7 @@ def __init__(self, *args, **kwargs):
 
     wait_time = between(1, 2)
 
-    all_text_queries = [
+    text_queries = [
         'travel with plane',
         'travel',
         'horse',
@@ -39,7 +39,7 @@ def __init__(self, *args, **kwargs):
 
     @task
     def search(self):
-        self.client.index(INDEX_NAME).search(q=random.choice(self.all_text_queries))
+        self.client.index(INDEX_NAME).search(q=random.choice(self.image_queries))
 
 
 @events.quitting.add_listener