Fix rerank benchmark script (#107)

gavinlichn · web-flow · commit 8edda1cc3872 · 2024-09-05T23:00:40.000+08:00
Signed-off-by: Li Gang &lt;gang.g.li@intel.com&gt;
diff --git a/evals/benchmark/stresscli/locust/rerankingfixed.py b/evals/benchmark/stresscli/locust/rerankingfixed.py
@@ -14,7 +14,7 @@ def getUrl():
 
 
 def getReqData():
-    return ({"initial_query": my_query, "retrieved_docs": [{"text": query_rerank_1}, {"text": query_rerank_2}]},)
+    return {"initial_query": my_query, "retrieved_docs": [{"text": query_rerank_1}, {"text": query_rerank_2}]}
 
 
 def respStatics(environment, resp):
diff --git a/evals/benchmark/stresscli/locust/rerankservefixed.py b/evals/benchmark/stresscli/locust/rerankservefixed.py
@@ -14,7 +14,7 @@ def getUrl():
 
 
 def getReqData():
-    return ({"query": my_query, "texts": [query_rerank_1, query_rerank_2]},)
+    return {"query": my_query, "texts": [query_rerank_1, query_rerank_2]}
 
 
 def respStatics(environment, resp):
