diff --git a/.github/workflows/vllm_ascend_test.yaml b/.github/workflows/vllm_ascend_test.yaml index 048cd49..60f04d3 100644 --- a/.github/workflows/vllm_ascend_test.yaml +++ b/.github/workflows/vllm_ascend_test.yaml @@ -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 diff --git a/tests/ut/torchair/test_utils.py b/tests/ut/torchair/test_utils.py index edd3fc2..8965738 100644 --- a/tests/ut/torchair/test_utils.py +++ b/tests/ut/torchair/test_utils.py @@ -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",