From 0ed3186db114700e058687eceaca45e9dee13604 Mon Sep 17 00:00:00 2001 From: qiye Date: Fri, 22 May 2026 11:32:14 +0800 Subject: [PATCH] [Bugfix] Remove stale OmniStage import and type annotation (#3541) Signed-off-by: Jianliang Qi --- vllm_omni/entrypoints/pd_utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/vllm_omni/entrypoints/pd_utils.py b/vllm_omni/entrypoints/pd_utils.py index 413d5d6b4..bdc721ab0 100644 --- a/vllm_omni/entrypoints/pd_utils.py +++ b/vllm_omni/entrypoints/pd_utils.py @@ -13,8 +13,6 @@ from typing import TYPE_CHECKING, Any if TYPE_CHECKING: from vllm import SamplingParams - from vllm_omni.entrypoints.omni_stage import OmniStage - _DEFAULT_MOONCAKE_BOOTSTRAP_PORT = 25201 logger = logging.getLogger(__name__) @@ -131,7 +129,7 @@ class PDDisaggregationMixin: p_stage = self.stage_configs[p_id] d_stage = self.stage_configs[d_id] - def _get_kv_cfg(stage: "OmniStage") -> dict[str, Any]: + def _get_kv_cfg(stage: Any) -> dict[str, Any]: ea = stage.engine_args cfg = getattr(ea, "kv_transfer_config", None) if cfg is None: