Format code & move functions (#155)

2024-02-06 13:27:46 -08:00
parent a7334aeea1
commit 23f05005fd
14 changed files with 94 additions and 54 deletions
--- a/test/srt/test_openai_server.py
+++ b/test/srt/test_openai_server.py
@@ -64,9 +64,8 @@ def test_completion_stream(args, echo, logprobs):
            first = False
        if logprobs:
            print(
-                f"{r.choices[0].text:12s}\t"
-                f"{r.choices[0].logprobs.token_logprobs}",
-                flush=True
+                f"{r.choices[0].text:12s}\t" f"{r.choices[0].logprobs.token_logprobs}",
+                flush=True,
            )
        else:
            print(r.choices[0].text, end="", flush=True)