Unverified Commit 7f1f36bf authored by Martin Hickey's avatar Martin Hickey Committed by GitHub
Browse files

[CI] Fix mypy for vllm/reasoning (#35742)


Signed-off-by: default avatarMartin Hickey <martin.hickey@ie.ibm.com>
Signed-off-by: default avatarHarry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: default avatarHarry Mellor <19981378+hmellor@users.noreply.github.com>
parent 5282c7d4
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
from collections.abc import Sequence
from pathlib import Path
from typing import TYPE_CHECKING, Any, Protocol, overload
......@@ -116,12 +117,14 @@ class TokenizerLike(Protocol):
def convert_tokens_to_string(self, tokens: list[str]) -> str:
raise NotImplementedError
def decode(self, ids: list[int] | int, skip_special_tokens: bool = False) -> str:
def decode(
self, ids: Sequence[int] | int, skip_special_tokens: bool = False
) -> str:
raise NotImplementedError
def convert_ids_to_tokens(
self,
ids: list[int],
ids: Sequence[int],
skip_special_tokens: bool = False,
) -> list[str]:
raise NotImplementedError
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment