mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-04-03 09:17:03 +08:00
fix mypy
Signed-off-by: inkcherry <mingzhi.liu@amd.com>
This commit is contained in:
parent
9b90f5ddb2
commit
4c79f34e8a
@ -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),
|
||||
)
|
||||
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user