From bb064e243fd632089c5b06f4b87d06a007bd2c25 Mon Sep 17 00:00:00 2001 From: wt Date: Mon, 1 Nov 2021 18:06:03 +0800 Subject: [PATCH] [skip ci] Update codes of benchmark (#11016) Signed-off-by: wangting0128 --- tests/benchmark/milvus_benchmark/runners/accuracy.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/benchmark/milvus_benchmark/runners/accuracy.py b/tests/benchmark/milvus_benchmark/runners/accuracy.py index aedd3e0b2..450421668 100644 --- a/tests/benchmark/milvus_benchmark/runners/accuracy.py +++ b/tests/benchmark/milvus_benchmark/runners/accuracy.py @@ -265,6 +265,13 @@ class AccAccuracyRunner(AccuracyRunner): true_ids = case_param["true_ids"] nq = case_metric.search["nq"] top_k = case_metric.search["topk"] + start_time = time.time() + end_time = start_time + 500 + cnt = 0 + while cnt < 100 and start_time < end_time: + self.milvus.query(case_param["vector_query"], filter_query=case_param["filter_query"]) + cnt += 1 + start_time = time.time() query_res = self.milvus.query(case_param["vector_query"], filter_query=case_param["filter_query"]) result_ids = self.milvus.get_ids(query_res) # Calculate the accuracy of the result of query -- GitLab