From 9a5436dca49ba6dde3df918a2c68f7d093c3d85d Mon Sep 17 00:00:00 2001
From: wt <ting.wang@zilliz.com>
Date: Wed, 10 Nov 2021 19:35:50 +0800
Subject: [PATCH] [skip ci] Update codes of StepLoadShape function (#11543)

Signed-off-by: wangting0128 <ting.wang@zilliz.com>
---
 tests/benchmark/milvus_benchmark/runners/locust_user.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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):