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-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
from collections.abc import Sequence
from pathlib import Path from pathlib import Path
from typing import TYPE_CHECKING, Any, Protocol, overload from typing import TYPE_CHECKING, Any, Protocol, overload
...@@ -116,12 +117,14 @@ class TokenizerLike(Protocol): ...@@ -116,12 +117,14 @@ class TokenizerLike(Protocol):
def convert_tokens_to_string(self, tokens: list[str]) -> str: def convert_tokens_to_string(self, tokens: list[str]) -> str:
raise NotImplementedError 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 raise NotImplementedError
def convert_ids_to_tokens( def convert_ids_to_tokens(
self, self,
ids: list[int], ids: Sequence[int],
skip_special_tokens: bool = False, skip_special_tokens: bool = False,
) -> list[str]: ) -> list[str]:
raise NotImplementedError 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