Skip to content

Commit e249b4c

Browse files
committed
mock nixl
Signed-off-by: NickLucche <[email protected]>
1 parent 772e7f4 commit e249b4c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/v1/kv_connector/unit/test_nixl_connector.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -376,6 +376,9 @@ def test_concurrent_load_kv(
376376
raise TimeoutError("Took too long to complete async handshake.")
377377

378378

379+
@patch(
380+
"vllm.distributed.kv_transfer.kv_connector.v1.nixl_connector.NixlWrapper",
381+
FakeNixlWrapper)
379382
def test_abort_timeout_on_prefiller(monkeypatch):
380383
"""
381384
Test lifecycle of an aborted Remote Prefill request hitting the timeout.

0 commit comments

Comments
 (0)