From b26a9735e6211776dea8a6c8b9c07d6e3574f2aa Mon Sep 17 00:00:00 2001 From: tzhouam Date: Wed, 28 Jan 2026 15:41:20 +0800 Subject: [PATCH] [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] Signed-off-by: tzhouam --- vllm_omni/engine/arg_utils.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/vllm_omni/engine/arg_utils.py b/vllm_omni/engine/arg_utils.py index cbe2001d7..222f069c4 100644 --- a/vllm_omni/engine/arg_utils.py +++ b/vllm_omni/engine/arg_utils.py @@ -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