diff --git a/tests/benchmark/milvus_benchmark/runners/locust_tasks.py b/tests/benchmark/milvus_benchmark/runners/locust_tasks.py index 5bd6c7f1d8..e35d95517c 100644 --- a/tests/benchmark/milvus_benchmark/runners/locust_tasks.py +++ b/tests/benchmark/milvus_benchmark/runners/locust_tasks.py @@ -77,7 +77,7 @@ class Tasks(TaskSet): """ query interface """ op = "get" # ids = [random.randint(1, 10000000) for _ in range(self.params[op]["ids_length"])] - self.client.get(self.values["get_ids"][:self.params[op]["ids_length"]]) + self.client.get(self.values["get_ids"][:self.params[op]["ids_length"]], timeout=300) @task def scene_test(self):