Skip to content

Commit ad27bb9

Browse files
author
angazenn
committed
fix ut
Signed-off-by: angazenn <zengyanjia@huawei.com>
1 parent 41618ed commit ad27bb9

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

tests/ut/test_platform.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -503,9 +503,10 @@ def test_get_attn_backend_cls_use_v1_and_mla(self, mock_get_ascend_config):
503503
)
504504
self.assertEqual(result,
505505
"vllm_ascend.attention.mla_v1.AscendMLABackend")
506-
506+
507507
@patch('vllm_ascend.platform.get_ascend_config')
508-
def test_get_attn_backend_cls_use_v1_and_torchair(self, mock_get_ascend_config):
508+
def test_get_attn_backend_cls_use_v1_and_torchair(self,
509+
mock_get_ascend_config):
509510
mock_config = MagicMock()
510511
mock_config.torchair_graph_config.enabled = True
511512

@@ -522,7 +523,8 @@ def test_get_attn_backend_cls_use_v1_and_torchair(self, mock_get_ascend_config):
522523
)
523524
self.assertEqual(
524525
result,
525-
"vllm_ascend.attention.attention_v1_torchair.AscendAttentionTorchairBackend")
526+
"vllm_ascend.attention.attention_v1_torchair.AscendAttentionTorchairBackend"
527+
)
526528

527529
@patch('vllm_ascend.platform.get_ascend_config')
528530
def test_get_attn_backend_cls_use_v1_only(self, mock_get_ascend_config):

0 commit comments

Comments
 (0)