From 9435e37bbd75d2f92df7f9c8561c62a4e49f1f08 Mon Sep 17 00:00:00 2001 From: wt Date: Wed, 8 Dec 2021 13:35:13 +0800 Subject: [PATCH] [skip ci] Update tasks of locust on benchmark (#12961) Signed-off-by: wangting0128 --- tests/benchmark/milvus_benchmark/runners/locust_user.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/benchmark/milvus_benchmark/runners/locust_user.py b/tests/benchmark/milvus_benchmark/runners/locust_user.py index ac935ded47..9a6fddfc14 100644 --- a/tests/benchmark/milvus_benchmark/runners/locust_user.py +++ b/tests/benchmark/milvus_benchmark/runners/locust_user.py @@ -55,13 +55,15 @@ class MyUser(User): def locust_executor(host, port, collection_name, connection_type="single", run_params=None): m = MilvusClient(host=host, port=port, collection_name=collection_name) - MyUser.tasks = {} + # MyUser.tasks = {} MyUser.op_info = run_params["op_info"] MyUser.params = {} tasks = run_params["tasks"] for op, value in tasks.items(): - task = {eval("Tasks." + op): value["weight"]} - MyUser.tasks.update(task) + # task = {eval("Tasks." + op): value["weight"]} + # MyUser.tasks.update(task) + for i in range(int(value["weight"])): + MyUser.tasks.append(eval("Tasks." + op)) MyUser.params[op] = value["params"] if "params" in value else None logger.info(MyUser.tasks)