From 02c4f3aa0f88d1f81eafa36fc65e1b8e85ea14b8 Mon Sep 17 00:00:00 2001 From: wt Date: Fri, 15 Oct 2021 15:30:42 +0800 Subject: [PATCH] [skip ci] Update search interface imput params on benchmark (#9950) Signed-off-by: wangting0128 --- tests/benchmark/milvus_benchmark/client.py | 19 ++++++++++++++++--- 1 file changed, 16 insertions(+), 3 deletions(-) diff --git a/tests/benchmark/milvus_benchmark/client.py b/tests/benchmark/milvus_benchmark/client.py index 9e1fffde25..1d137c5a1f 100644 --- a/tests/benchmark/milvus_benchmark/client.py +++ b/tests/benchmark/milvus_benchmark/client.py @@ -311,7 +311,12 @@ class MilvusClient(object): query = { "bool": {"must": must_params} } - result = self._milvus.search(tmp_collection_name, query, timeout=timeout) + + params = util.search_param_analysis(vector_query, filter_query) + params.update({"timeout": timeout}) + result = self._milvus.search(tmp_collection_name, **params) + + # result = self._milvus.search(tmp_collection_name, query, timeout=timeout) return result @time_wrapper @@ -330,7 +335,10 @@ class MilvusClient(object): } logger.debug("Start warm up query") for i in range(times): - self._milvus.search(self._collection_name, query) + params = util.search_param_analysis(vector_query, None) + result = self._milvus.search(self._collection_name, **params) + + # self._milvus.search(self._collection_name, query) logger.debug("End warm up query") @time_wrapper @@ -343,7 +351,12 @@ class MilvusClient(object): "bool": {"must": must_params} } self.load_collection(tmp_collection_name) - result = self._milvus.search(tmp_collection_name, query, timeout=timeout) + + params = util.search_param_analysis(vector_query, filter_query) + params.update({"timeout": timeout}) + result = self._milvus.search(tmp_collection_name, **params) + + # result = self._milvus.search(tmp_collection_name, query, timeout=timeout) return result def get_ids(self, result):