From 4c79f34e8aa73e73d7a66a0772dbcaf4420e62a0 Mon Sep 17 00:00:00 2001 From: inkcherry Date: Thu, 20 Nov 2025 11:36:31 +0000 Subject: [PATCH] fix mypy Signed-off-by: inkcherry --- .../kv_transfer/kv_connector/v1/moriio_connector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vllm/distributed/kv_transfer/kv_connector/v1/moriio_connector.py b/vllm/distributed/kv_transfer/kv_connector/v1/moriio_connector.py index a3737700d20f3..2e641e451319d 100644 --- a/vllm/distributed/kv_transfer/kv_connector/v1/moriio_connector.py +++ b/vllm/distributed/kv_transfer/kv_connector/v1/moriio_connector.py @@ -1678,7 +1678,7 @@ class MoRIIOConnectorWorker: remote_tp_size: int, expected_engine_id: str, remote_dp_rank: int = 0, - ) -> set[str]: + ) -> dict[int, str]: """Do a MoRIIO handshake with a remote instance.""" start_time = time.perf_counter() @@ -2263,7 +2263,7 @@ class MoRIIOConnectorWorker: self._recving_transfers[request_id].append(transfer_status) self._recving_transfers_callback_addr[request_id] = ( remote_host, - remote_notify_port + self.tp_rank, + str(remote_notify_port + self.tp_rank), )