[Enhancement] Update type hint for stage_connector_spec in OmniEngineArgs and AsyncOmniEngineArgs

- Changed type hint for stage_connector_spec from 'any' to 'Any' for improved type checking and consistency.
- Ensured compatibility with typing standards in the arg_utils module.

Signed-off-by: [Your Name] <your.email@example.com>

Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
tzhouam committed 2026-01-28 15:41:20 +08:00
1 parent d7cfe195ea
commit b26a9735e6
1 file changed
+3 -2
+3 -2
View File
@@ -1,4 +1,5 @@
from dataclasses import dataclass, field
from typing import Any
from transformers.models.qwen3_omni_moe.configuration_qwen3_omni_moe import Qwen3OmniMoeTextConfig
from vllm.engine.arg_utils import EngineArgs
@@ -69,7 +70,7 @@ class OmniEngineArgs(EngineArgs):
engine_output_type: str | None = None
hf_config_name: str | None = None
custom_process_next_stage_input_func: str | None = None
stage_connector_spec: dict[str, any] = field(default_factory=dict)
stage_connector_spec: dict[str, Any] = field(default_factory=dict)
async_chunk: bool = False
omni_kv_config: dict | None = None
@@ -167,7 +168,7 @@ class AsyncOmniEngineArgs(AsyncEngineArgs):
engine_output_type: str | None = None
hf_config_name: str | None = None
custom_process_next_stage_input_func: str | None = None
stage_connector_spec: dict[str, any] = field(default_factory=dict)
stage_connector_spec: dict[str, Any] = field(default_factory=dict)
async_chunk: bool = False
omni_kv_config: dict | None = None