Skip to content

Commit e535a51

Browse files
committed
fix: make lint happy
Signed-off-by: zongz <zongzhe1024@163.com>
1 parent ceb6abd commit e535a51

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/e2e/multicard/test_qwen3_moe.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,7 @@ def test_models_distributed_Qwen3_MOE_TP2_WITH_ACLGRAPH():
102102
) as vllm_model:
103103
vllm_model.generate_greedy(example_prompts, max_tokens)
104104

105+
105106
def test_Qwen3_235b_all2allv_mc2_quant(monkeypatch):
106107
"""Test Qwen3-235B with all2all sequence and multi-card configuration."""
107108
# Set environment variables similar to the startup command
@@ -113,14 +114,14 @@ def test_Qwen3_235b_all2allv_mc2_quant(monkeypatch):
113114
monkeypatch.setenv('ACL_STREAM_TIMEOUT', '340000')
114115
monkeypatch.setenv('HCCL_OP_EXPANSION_MODE', 'AIV')
115116
monkeypatch.setenv('HCCL_OP_BASE_FFTS_MODE_ENABLE', 'true')
116-
117+
117118
example_prompts = [
118119
"Hello, my name is",
119120
"The capital of France is",
120121
"In the field of artificial intelligence,",
121122
]
122123
max_tokens = 32
123-
124+
124125
# Additional config matching the startup command
125126
additional_config = {
126127
"torchair_graph_config": {
@@ -134,7 +135,7 @@ def test_Qwen3_235b_all2allv_mc2_quant(monkeypatch):
134135
},
135136
"refresh": True
136137
}
137-
138+
138139
with VllmRunner(
139140
"vllm-ascend/Qwen3-235B-A22B-W8A8", # Use quantized model path
140141
tensor_parallel_size=4,
@@ -150,4 +151,3 @@ def test_Qwen3_235b_all2allv_mc2_quant(monkeypatch):
150151
additional_config=additional_config,
151152
) as vllm_model:
152153
vllm_model.generate_greedy(example_prompts, max_tokens)
153-

0 commit comments

Comments
 (0)