minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[CI][RFC] Replace black-jupyter with ruff-format (#37210)

Browse Source 
Co-authored-by: Alison Shao <a.shao@wustl.edu>
This commit is contained in:
Alex Nails
2026-09-02 19:46:08 -07:00
committed by GitHub
co-authored by Alison Shao
parent 2641e427be
commit 28262c20df
1411 changed files with 7766 additions and 8176 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/manual/test_sagemaker_server.py
+9 -9
View File
@@ -69,9 +69,9 @@ class TestSageMakerServer(CustomTestCase):
ret_num_top_logprobs = len(
response["choices"][0]["logprobs"]["content"][0]["top_logprobs"]
)
assert (
ret_num_top_logprobs == logprobs
), f"{ret_num_top_logprobs} vs {logprobs}"
assert ret_num_top_logprobs == logprobs, (
f"{ret_num_top_logprobs} vs {logprobs}"
)
assert len(response["choices"]) == parallel_sample_num
assert response["choices"][0]["message"]["role"] == "assistant"
@@ -155,18 +155,18 @@ class TestSageMakerServer(CustomTestCase):
.get("content")[0]
.get("top_logprobs")
)
assert (
ret_num_top_logprobs == logprobs
), f"{ret_num_top_logprobs} vs {logprobs}"
assert ret_num_top_logprobs == logprobs, (
f"{ret_num_top_logprobs} vs {logprobs}"
)
assert isinstance(data["content"], str)
assert line["id"]
assert line["created"]
for index in [i for i in range(parallel_sample_num)]:
assert not is_firsts.get(
index, True
), f"index {index} is not found in the response"
assert not is_firsts.get(index, True), (
f"index {index} is not found in the response"
)
def test_chat_completion(self):
for logprobs in [None, 5]: