xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-09 20:15:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Docs] Fix griffe warning in base_static_graph.py (#25018)

Browse Source 
Signed-off-by: windsonsea <haifeng.yao@daocloud.io>
This commit is contained in:
Michael Yao 2025-09-17 16:49:19 +08:00 committed by GitHub
parent 6c47f6bfa4
commit 0fb2551c23
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
vllm/compilation/base_static_graph.py
View File

@ -12,8 +12,13 @@ class AbstractStaticGraphWrapper(Protocol):
to be captured as a static graph.
"""
def __init__(self, runnable: Callable, vllm_config: VllmConfig,
runtime_mode: CUDAGraphMode, **kwargs):
def __init__(
self,
runnable: Callable[..., Any],
vllm_config: VllmConfig,
runtime_mode: CUDAGraphMode,
**kwargs: Any,
) -> None:
"""
Initializes the StaticGraphWrapper class with graph capturing and
execution-related configurations.
@ -31,7 +36,7 @@ class AbstractStaticGraphWrapper(Protocol):
"""
raise NotImplementedError
def __call__(self, *args, **kwargs) -> Any:
def __call__(self, *args: Any, **kwargs: Any) -> Any:
"""
Executes the wrapped callable.