Format (#593)

dc1b8bcf · Ying Sheng · GitHub · 5a57b8ad · dc1b8bcf
Unverified Commit dc1b8bcf authored Jul 05, 2024 by Ying Sheng Committed by GitHub Jul 05, 2024
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 5 deletions

test/srt/test_httpserver_classify.py test/srt/test_httpserver_classify.py +7 -5

No files found.
--- a/test/srt/test_httpserver_classify.py
+++ b/test/srt/test_httpserver_classify.py
@@ -37,10 +37,12 @@ def get_logits_batch(url, prompts):
        },
    )
    ret = response.json()
-    logits = np.array(list(
+    logits = np.array(
+        list(
            ret[i]["meta_info"]["normalized_prompt_logprob"]
            for i in range(len(prompts))
-    ))
+        )
+    )
    return logits