[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:
menogrey
2025-10-14 08:31:13 +08:00
committed by GitHub
parent 4f6d60eb06
commit 657c08cfb2
2 changed files with 4 additions and 2 deletions

View File

@@ -128,7 +128,6 @@ jobs:
--ignore=tests/ut/kv_connector/test_remote_decode_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/test_utils.py
- name: Upload coverage to Codecov
# only upload coverage when commits merged

View File

@@ -64,7 +64,7 @@ class TestTorchairUtils(TestBase):
mock_model_registry.return_value = mock_registry
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
expected_registrations = [
@@ -77,6 +77,9 @@ class TestTorchairUtils(TestBase):
("DeepseekV3ForCausalLM",
"vllm_ascend.torchair.models.torchair_deepseek_v3:TorchairDeepseekV3ForCausalLM"
),
("DeepseekV32ForCausalLM",
"vllm_ascend.torchair.models.torchair_deepseek_v3:TorchairDeepseekV3ForCausalLM"
),
("Qwen2ForCausalLM",
"vllm_ascend.torchair.models.qwen2:CustomQwen2ForCausalLM"),
("Qwen3MoeForCausalLM",