[UT] fix skipped test_utils ut test. (#3422)
### What this PR does / why we need it? Fixes: fix the test in `tests/ut/torchair/test_utils.py` and enable the UT test in CI. ### Does this PR introduce _any_ user-facing change? No. ### How was this patch tested? vLLM version: v0.11.0rc3 vLLM main: https://github.com/vllm-project/vllm/commit/v0.11.0 - vLLM version: v0.11.0rc3 - vLLM main: https://github.com/vllm-project/vllm/commit/v0.11.0 Signed-off-by: menogrey <1299267905@qq.com>
This commit is contained in:
1
.github/workflows/vllm_ascend_test.yaml
vendored
1
.github/workflows/vllm_ascend_test.yaml
vendored
@@ -128,7 +128,6 @@ jobs:
|
|||||||
--ignore=tests/ut/kv_connector/test_remote_decode_lifecycle.py \
|
--ignore=tests/ut/kv_connector/test_remote_decode_lifecycle.py \
|
||||||
--ignore=tests/ut/kv_connector/test_remote_prefill_lifecycle.py \
|
--ignore=tests/ut/kv_connector/test_remote_prefill_lifecycle.py \
|
||||||
--ignore=tests/ut/torchair/models/test_torchair_deepseek_v2.py \
|
--ignore=tests/ut/torchair/models/test_torchair_deepseek_v2.py \
|
||||||
--ignore=tests/ut/torchair/test_utils.py
|
|
||||||
|
|
||||||
- name: Upload coverage to Codecov
|
- name: Upload coverage to Codecov
|
||||||
# only upload coverage when commits merged
|
# only upload coverage when commits merged
|
||||||
|
|||||||
@@ -64,7 +64,7 @@ class TestTorchairUtils(TestBase):
|
|||||||
mock_model_registry.return_value = mock_registry
|
mock_model_registry.return_value = mock_registry
|
||||||
utils.register_torchair_model()
|
utils.register_torchair_model()
|
||||||
|
|
||||||
self.assertEqual(mock_model_registry.register_model.call_count, 6)
|
self.assertEqual(mock_model_registry.register_model.call_count, 7)
|
||||||
call_args_list = mock_model_registry.register_model.call_args_list
|
call_args_list = mock_model_registry.register_model.call_args_list
|
||||||
|
|
||||||
expected_registrations = [
|
expected_registrations = [
|
||||||
@@ -77,6 +77,9 @@ class TestTorchairUtils(TestBase):
|
|||||||
("DeepseekV3ForCausalLM",
|
("DeepseekV3ForCausalLM",
|
||||||
"vllm_ascend.torchair.models.torchair_deepseek_v3:TorchairDeepseekV3ForCausalLM"
|
"vllm_ascend.torchair.models.torchair_deepseek_v3:TorchairDeepseekV3ForCausalLM"
|
||||||
),
|
),
|
||||||
|
("DeepseekV32ForCausalLM",
|
||||||
|
"vllm_ascend.torchair.models.torchair_deepseek_v3:TorchairDeepseekV3ForCausalLM"
|
||||||
|
),
|
||||||
("Qwen2ForCausalLM",
|
("Qwen2ForCausalLM",
|
||||||
"vllm_ascend.torchair.models.qwen2:CustomQwen2ForCausalLM"),
|
"vllm_ascend.torchair.models.qwen2:CustomQwen2ForCausalLM"),
|
||||||
("Qwen3MoeForCausalLM",
|
("Qwen3MoeForCausalLM",
|
||||||
|
|||||||
Reference in New Issue
Block a user