From 3092b29eeaedb284b812a7b13924d3207da0a226 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 24 Oct 2024 09:58:49 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- evals/benchmark/stresscli/locust/tokenresponse.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/evals/benchmark/stresscli/locust/tokenresponse.py b/evals/benchmark/stresscli/locust/tokenresponse.py index b566c0a1..77f9740b 100644 --- a/evals/benchmark/stresscli/locust/tokenresponse.py +++ b/evals/benchmark/stresscli/locust/tokenresponse.py @@ -26,7 +26,9 @@ def respStatics(environment, req, resp): num_token_input_prompt = -1 num_token_output = len( - tokenizer.encode(resp["response_string"].lstrip().encode("utf-8").decode("unicode_escape"), add_special_tokens=False) + tokenizer.encode( + resp["response_string"].lstrip().encode("utf-8").decode("unicode_escape"), add_special_tokens=False + ) ) return {