From 13560392d2571623e0f4050a400dded0432e4dc6 Mon Sep 17 00:00:00 2001 From: Cyrus Leung Date: Wed, 22 Jan 2025 12:06:54 +0800 Subject: [PATCH] [Ci/Build] Fix mypy errors on main (#12296) Signed-off-by: DarkLight1337 --- vllm/multimodal/processing.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/vllm/multimodal/processing.py b/vllm/multimodal/processing.py index a0993f377dea7..1e0338dfb0226 100644 --- a/vllm/multimodal/processing.py +++ b/vllm/multimodal/processing.py @@ -43,7 +43,7 @@ class PromptReplacementDetails: """ @staticmethod - def from_seq(seq: _PromptSeq): + def from_seq(seq: _PromptSeq) -> "PromptReplacementDetails": return PromptReplacementDetails(full=seq, features=seq) @@ -132,7 +132,10 @@ class _BoundPromptSequence: _token_ids: Optional[list[int]] @staticmethod - def from_seq(tokenizer: AnyTokenizer, seq: _PromptSeq): + def from_seq( + tokenizer: AnyTokenizer, + seq: _PromptSeq, + ) -> "_BoundPromptSequence": return _BoundPromptSequence( tokenizer=tokenizer, _text=seq if isinstance(seq, str) else None,