Skip to content

Commit cb81376

Browse files
ut test
Signed-off-by: CodeNine-CJ <[email protected]>
1 parent 519ca7e commit cb81376

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/ut/torchair/test_torchair_model_runner.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ def setup_npu_torchair_model_runner(self, mocker: MockerFixture):
103103
vllm_config.scheduler_config = scheduler_config
104104

105105
vllm_config.load_config = MagicMock()
106-
106+
107107
vllm_config.kv_transfer_config = None
108108

109109
mocker.patch(
@@ -125,7 +125,6 @@ def setup_npu_torchair_model_runner(self, mocker: MockerFixture):
125125
"vllm_ascend.torchair.torchair_model_runner._may_pad_kv_consumer_num_seq"
126126
)
127127

128-
129128
device = torch.device("npu:0")
130129
ascend_config = MagicMock()
131130

0 commit comments

Comments
 (0)