diff --git a/tests/benchmark/milvus_benchmark/runners/locust_user.py b/tests/benchmark/milvus_benchmark/runners/locust_user.py index e7a19aa7f8..60f56f44da 100644 --- a/tests/benchmark/milvus_benchmark/runners/locust_user.py +++ b/tests/benchmark/milvus_benchmark/runners/locust_user.py @@ -45,7 +45,7 @@ class StepLoadShape(LoadTestShape): return None current_step = math.floor(run_time / self.step_time) + 1 - return (current_step * self.step_load, self.spawn_rate) + return current_step * self.step_load, self.spawn_rate class MyUser(User):