diff --git a/evals/benchmark/stresscli/locust/rerankingfixed.py b/evals/benchmark/stresscli/locust/rerankingfixed.py index d715c5f2..6c780291 100644 --- 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 index c5e4070e..342b44e2 100644 --- 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):