From f63cc19210f6c5eb5036a989632c60766aa70495 Mon Sep 17 00:00:00 2001 From: Nick Hill Date: Wed, 23 Jul 2025 08:08:40 +0100 Subject: [PATCH] fix handshake mock test Signed-off-by: Nick Hill --- tests/v1/engine/test_engine_core_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/v1/engine/test_engine_core_client.py b/tests/v1/engine/test_engine_core_client.py index 65f1da803fb2e..2ac6dc796bd10 100644 --- a/tests/v1/engine/test_engine_core_client.py +++ b/tests/v1/engine/test_engine_core_client.py @@ -565,8 +565,8 @@ def test_engine_core_proc_instantiation_cuda_empty( from vllm.v1.engine.utils import EngineZmqAddresses - def mock_startup_handshake(self, handshake_socket, on_head_node, - parallel_config): + def mock_startup_handshake(self, handshake_socket, local_client, + headless, parallel_config): return EngineZmqAddresses(inputs=["tcp://127.0.0.1:5555"], outputs=["tcp://127.0.0.1:5556"], coordinator_input=None,