[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
@@ -42,9 +42,9 @@ def test_mixer2_gated_norm_multi_gpu(
|
||||
if device not in ["cuda", "xpu"]:
|
||||
pytest.skip("Test only supports CUDA and XPU devices")
|
||||
|
||||
assert (
|
||||
get_device_count() >= NUM_GPUS
|
||||
), f"This test requires at least {NUM_GPUS} GPUs, but only {get_device_count()} available"
|
||||
assert get_device_count() >= NUM_GPUS, (
|
||||
f"This test requires at least {NUM_GPUS} GPUs, but only {get_device_count()} available"
|
||||
)
|
||||
|
||||
hidden_size, n_groups = hidden_size_n_groups
|
||||
num_processes = NUM_GPUS
|
||||
|
||||
Reference in New Issue
Block a user