From ee97722cd38c62f3cf10954c1ba73449ccf5bfbf Mon Sep 17 00:00:00 2001 From: wt Date: Wed, 8 Dec 2021 13:37:05 +0800 Subject: [PATCH] [skip ci] Add gt of locust query func on benchmark (#12963) Signed-off-by: wangting0128 --- tests/benchmark/milvus_benchmark/runners/locust_tasks.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/benchmark/milvus_benchmark/runners/locust_tasks.py b/tests/benchmark/milvus_benchmark/runners/locust_tasks.py index 5eb0b17572..927eadf437 100644 --- a/tests/benchmark/milvus_benchmark/runners/locust_tasks.py +++ b/tests/benchmark/milvus_benchmark/runners/locust_tasks.py @@ -26,8 +26,12 @@ class Tasks(TaskSet): filter_query.append(eval(filter["range"])) if isinstance(filter, dict) and "term" in filter: filter_query.append(eval(filter["term"])) + + guarantee_timestamp = self.params[op]["guarantee_timestamp"] if "guarantee_timestamp" in self.params[op] else None + # logger.debug(filter_query) - self.client.query(vector_query, filter_query=filter_query, log=False, timeout=30) + self.client.query(vector_query, filter_query=filter_query, log=False, guarantee_timestamp=guarantee_timestamp, + timeout=30) @task def flush(self):