[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
opea-project · Dec 20, 2024 · 7ed73fe · 7ed73fe
1 parent 4c79f8a
commit 7ed73fe
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/evals/benchmark/stresscli/locust/aistress.py b/evals/benchmark/stresscli/locust/aistress.py
@@ -180,9 +180,9 @@ def bench_main(self):
                                     first_token_ts = time.perf_counter()
                                 try:
                                     data = json.loads(event.data)
-                                    for op in data['ops']:
-                                        if op['path'] == '/logs/HuggingFaceEndpoint/final_output':
-                                            generations = op['value'].get('generations', [])
+                                    for op in data["ops"]:
+                                        if op["path"] == "/logs/HuggingFaceEndpoint/final_output":
+                                            generations = op["value"].get("generations", [])
                                             for generation in generations:
                                                 for item in generation:
                                                     text = item.get("text", "")

diff --git a/evals/benchmark/stresscli/locust/faqgenfixed.py b/evals/benchmark/stresscli/locust/faqgenfixed.py
@@ -10,7 +10,7 @@ def getUrl():
 
 def getReqData():
     return {
-        "messages": "Text Embeddings Inference (TEI) is a toolkit for deploying and serving open source text embeddings and sequence classification models. TEI enables high-performance extraction for the most popular models, including FlagEmbedding, Ember, GTE and E6.", 
+        "messages": "Text Embeddings Inference (TEI) is a toolkit for deploying and serving open source text embeddings and sequence classification models. TEI enables high-performance extraction for the most popular models, including FlagEmbedding, Ember, GTE and E6.",
         "max_tokens": 128,
         "top_k": 1,
     }