Skip to content

Commit c4bd5e0

Browse files
windsonseaxuebwang-amd
authored andcommitted
[Docs] Fix griffe warning in base_static_graph.py (vllm-project#25018)
Signed-off-by: windsonsea <[email protected]> Signed-off-by: xuebwang-amd <[email protected]>
1 parent a065959 commit c4bd5e0

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

vllm/compilation/base_static_graph.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,13 @@ class AbstractStaticGraphWrapper(Protocol):
1212
to be captured as a static graph.
1313
"""
1414

15-
def __init__(self, runnable: Callable, vllm_config: VllmConfig,
16-
runtime_mode: CUDAGraphMode, **kwargs):
15+
def __init__(
16+
self,
17+
runnable: Callable[..., Any],
18+
vllm_config: VllmConfig,
19+
runtime_mode: CUDAGraphMode,
20+
**kwargs: Any,
21+
) -> None:
1722
"""
1823
Initializes the StaticGraphWrapper class with graph capturing and
1924
execution-related configurations.
@@ -31,7 +36,7 @@ def __init__(self, runnable: Callable, vllm_config: VllmConfig,
3136
"""
3237
raise NotImplementedError
3338

34-
def __call__(self, *args, **kwargs) -> Any:
39+
def __call__(self, *args: Any, **kwargs: Any) -> Any:
3540
"""
3641
Executes the wrapped callable.
3742

0 commit comments

Comments
 (0)