[mypy] Fix wrong type annotations related to tuple (#25660)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
2025-09-25 21:00:45 +08:00
parent 1e9a77e037
commit 2f17117606
9 changed files with 25 additions and 20 deletions
--- a/vllm/distributed/device_communicators/ray_communicator.py
+++ b/vllm/distributed/device_communicators/ray_communicator.py
@@ -178,7 +178,7 @@ class RayPPCommunicator(Communicator):

    def recv(
        self,
-        shape: tuple[int],
+        shape: tuple[int, ...],
        dtype: "torch.dtype",
        peer_rank: int,
        allocator: TorchTensorAllocator,