Skip to content

Commit 8edda1c

Browse files
authored
Fix rerank benchmark script (#107)
Signed-off-by: Li Gang <[email protected]>
1 parent 58502c5 commit 8edda1c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

evals/benchmark/stresscli/locust/rerankingfixed.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ def getUrl():
1414

1515

1616
def getReqData():
17-
return ({"initial_query": my_query, "retrieved_docs": [{"text": query_rerank_1}, {"text": query_rerank_2}]},)
17+
return {"initial_query": my_query, "retrieved_docs": [{"text": query_rerank_1}, {"text": query_rerank_2}]}
1818

1919

2020
def respStatics(environment, resp):

evals/benchmark/stresscli/locust/rerankservefixed.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ def getUrl():
1414

1515

1616
def getReqData():
17-
return ({"query": my_query, "texts": [query_rerank_1, query_rerank_2]},)
17+
return {"query": my_query, "texts": [query_rerank_1, query_rerank_2]}
1818

1919

2020
def respStatics(environment, resp):

0 commit comments

Comments
 (0)