Skip to content

Commit 57f5ba5

Browse files
committed
Fix format
Signed-off-by: Barry Kang <[email protected]>
1 parent 34cd672 commit 57f5ba5

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/integration/defs/test_e2e.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2183,6 +2183,7 @@ def test_ptp_quickstart_advanced_deepseek_r1_8gpus(llm_root, llm_venv,
21832183
stdout=running_log)
21842184
_check_mem_usage(running_log, [106.3, 0, 0, 0], 8)
21852185

2186+
21862187
@pytest.mark.skip_less_device_memory(110000)
21872188
@pytest.mark.skip_less_device(8)
21882189
@pytest.mark.parametrize("model_name,model_path", [
@@ -2220,7 +2221,8 @@ def test_relaxed_acceptance_quickstart_advanced_deepseek_r1_8gpus(
22202221
"--relaxed_delta=0.5",
22212222
"--enable_attention_dp",
22222223
"--use_one_model",
2223-
f"--moe_backend={"DEEPGEMM" if is_blackwell else "CUTLASS"}",
2224+
"--moe_backend",
2225+
"DEEPGEMM" if is_blackwell else "CUTLASS",
22242226
],
22252227
stdout=running_log)
22262228
_check_mem_usage(running_log, [85.6, 0, 0, 0], 8)

0 commit comments

Comments
 (0)