Unverified Commit 0fb2551c authored by Michael Yao's avatar Michael Yao Committed by GitHub
Browse files

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


Signed-off-by: default avatarwindsonsea <haifeng.yao@daocloud.io>
parent 6c47f6bf
...@@ -12,8 +12,13 @@ class AbstractStaticGraphWrapper(Protocol): ...@@ -12,8 +12,13 @@ class AbstractStaticGraphWrapper(Protocol):
to be captured as a static graph. to be captured as a static graph.
""" """
def __init__(self, runnable: Callable, vllm_config: VllmConfig, def __init__(
runtime_mode: CUDAGraphMode, **kwargs): self,
runnable: Callable[..., Any],
vllm_config: VllmConfig,
runtime_mode: CUDAGraphMode,
**kwargs: Any,
) -> None:
""" """
Initializes the StaticGraphWrapper class with graph capturing and Initializes the StaticGraphWrapper class with graph capturing and
execution-related configurations. execution-related configurations.
...@@ -31,7 +36,7 @@ class AbstractStaticGraphWrapper(Protocol): ...@@ -31,7 +36,7 @@ class AbstractStaticGraphWrapper(Protocol):
""" """
raise NotImplementedError raise NotImplementedError
def __call__(self, *args, **kwargs) -> Any: def __call__(self, *args: Any, **kwargs: Any) -> Any:
""" """
Executes the wrapped callable. Executes the wrapped callable.
......
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