diff --git a/tests/benchmark/milvus_benchmark/runners/locust_tasks.py b/tests/benchmark/milvus_benchmark/runners/locust_tasks.py index e35d95517c..e44a509e89 100644 --- a/tests/benchmark/milvus_benchmark/runners/locust_tasks.py +++ b/tests/benchmark/milvus_benchmark/runners/locust_tasks.py @@ -57,7 +57,7 @@ class Tasks(TaskSet): # ids = [random.randint(1000000, 10000000) for _ in range(self.params[op]["ni_per"])] # X = [[random.random() for _ in range(self.op_info["dimension"])] for _ in range(self.params[op]["ni_per"])] entities = utils.generate_entities(self.op_info["collection_info"], self.values["X"][:self.params[op]["ni_per"]], self.values["ids"][:self.params[op]["ni_per"]]) - self.client.insert(entities, log=False) + self.client.insert(entities, log=False, timeout=300) @task def insert_flush(self):