Skip to content

Commit 315a0e0

Browse files
committed
fix ut
Signed-off-by: Icey <1790571317@qq.com>
1 parent c4195b9 commit 315a0e0

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/ut/model_loader/netloader/test_netloader.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
from vllm_ascend.model_loader.netloader.netloader import ModelNetLoaderElastic
2525
from vllm_ascend.utils import vllm_version_is
2626

27+
2728
class DummyDeviceConfig:
2829
device = 'cuda'
2930
device_type = 'cuda'
@@ -176,7 +177,8 @@ def __exit__(self, a, b, c):
176177
if vllm_version_is("0.11.0"):
177178
monkeypatch.setattr("vllm.utils.get_ip", lambda: "127.0.0.1")
178179
else:
179-
monkeypatch.setattr("vllm.utils.network_utils.get_ip", lambda: "127.0.0.1")
180+
monkeypatch.setattr("vllm.utils.network_utils.get_ip",
181+
lambda: "127.0.0.1")
180182
# patch find_free_port
181183
monkeypatch.setattr(
182184
"vllm_ascend.model_loader.netloader.netloader.find_free_port",

0 commit comments

Comments
 (0)