Skip to content

Commit 8983e02

Browse files
authored
[CI] Fix Pre-commit Issue Cannot determine type of "rank" and "world_size" (#26448)
Signed-off-by: yewentao256 <zhyanwentao@126.com>
1 parent 1ee3538 commit 8983e02

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

vllm/distributed/device_communicators/all2all.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -377,6 +377,11 @@ class FlashInferAllToAllManager(All2AllManagerBase):
377377
All2All communication based on flashinfer kernels.
378378
"""
379379

380+
# This type lint could be removed after all of the work in
381+
# https://github.com/vllm-project/vllm/issues/26533 done.
382+
rank: int
383+
world_size: int
384+
380385
def __init__(self, cpu_group):
381386
assert has_flashinfer_all2all(), (
382387
"flashinfer all2all module not found. Please install/check flashinfer"

0 commit comments

Comments
 (0)