xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-24 02:44:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix mypy

Browse Source 
Signed-off-by: inkcherry <mingzhi.liu@amd.com>
This commit is contained in:
inkcherry 2025-11-20 11:36:31 +00:00
parent 9b90f5ddb2
commit 4c79f34e8a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vllm/distributed/kv_transfer/kv_connector/v1/moriio_connector.py
View File

@ -1678,7 +1678,7 @@ class MoRIIOConnectorWorker:
remote_tp_size: int, remote_tp_size: int,
expected_engine_id: str, expected_engine_id: str,
remote_dp_rank: int = 0, remote_dp_rank: int = 0,
) -> set[str]: ) -> dict[int, str]:
"""Do a MoRIIO handshake with a remote instance.""" """Do a MoRIIO handshake with a remote instance."""
start_time = time.perf_counter() start_time = time.perf_counter()
@ -2263,7 +2263,7 @@ class MoRIIOConnectorWorker:
self._recving_transfers[request_id].append(transfer_status) self._recving_transfers[request_id].append(transfer_status)
self._recving_transfers_callback_addr[request_id] = ( self._recving_transfers_callback_addr[request_id] = (
remote_host, remote_host,
remote_notify_port + self.tp_rank, str(remote_notify_port + self.tp_rank),
) )