From 4bdd2fbded7db673ce22b82d025a50985b683002 Mon Sep 17 00:00:00 2001 From: yihanzhao Date: Wed, 16 Oct 2024 10:41:20 +1100 Subject: [PATCH] test image inference only --- perf_tests/batch_add_doc_with_image.py | 2 +- perf_tests/search_with_image.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/perf_tests/batch_add_doc_with_image.py b/perf_tests/batch_add_doc_with_image.py index e17c36f48..1b197b8ee 100644 --- 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 index 4736c2b26..5c647c675 100644 --- 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