[CI][RFC] Replace black-jupyter with ruff-format (#37210)
Co-authored-by: Alison Shao <a.shao@wustl.edu>
This commit is contained in:
co-authored by
Alison Shao
parent
2641e427be
commit
28262c20df
@@ -411,10 +411,13 @@ class TestGetDcpLens(CustomTestCase):
|
||||
)
|
||||
# The allocator widens from get_parallel(), not from the injected
|
||||
# server_args stand-in -- drive the cause, not the effect.
|
||||
with patch(
|
||||
"sglang.srt.mem_cache.kv_cache_configurator.current_platform.is_out_of_tree",
|
||||
return_value=False,
|
||||
), rc.get_parallel().override(attn_dcp_size=dcp_size):
|
||||
with (
|
||||
patch(
|
||||
"sglang.srt.mem_cache.kv_cache_configurator.current_platform.is_out_of_tree",
|
||||
return_value=False,
|
||||
),
|
||||
rc.get_parallel().override(attn_dcp_size=dcp_size),
|
||||
):
|
||||
allocators[dcp_size] = (
|
||||
KVCacheConfigurator._build_token_to_kv_pool_allocator(
|
||||
configurator,
|
||||
|
||||
@@ -87,7 +87,7 @@ class TestQwen35TritonDCPGsm8k(CustomTestCase):
|
||||
|
||||
if is_in_ci():
|
||||
write_github_step_summary(
|
||||
f"### test_a_gsm8k (qwen3.5-triton-dcp4)\n" f'{metrics["score"]=:.3f}\n'
|
||||
f'### test_a_gsm8k (qwen3.5-triton-dcp4)\n{metrics["score"]=:.3f}\n'
|
||||
)
|
||||
self.assertGreater(metrics["score"], GSM8K_ACCURACY_THRESHOLD)
|
||||
|
||||
|
||||
@@ -85,8 +85,7 @@ def multiprocess_test(file: str, nproc: int, timeout: int = 120) -> None:
|
||||
) from e
|
||||
|
||||
assert result.returncode == 0, (
|
||||
f"torchrun (nproc={nproc}) failed with rc={result.returncode}\n"
|
||||
f"{result.stdout}"
|
||||
f"torchrun (nproc={nproc}) failed with rc={result.returncode}\n{result.stdout}"
|
||||
)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user