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

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
Cyrus Leung
2025-09-25 21:00:45 +08:00
committed by GitHub
parent 1e9a77e037
commit 2f17117606
9 changed files with 25 additions and 20 deletions

View File

@@ -1,6 +1,6 @@
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
from collections.abc import Sequence
from typing import Callable, Union
import torch
@@ -55,7 +55,7 @@ class NoBadWordsLogitsProcessor:
def __call__(
self,
past_tokens_ids: Union[list[int], tuple[int]],
past_tokens_ids: Sequence[int],
logits: torch.FloatTensor,
) -> torch.Tensor:
if self.word_bias is None: