From 4fab34c8f46f1f431571cf5c699f55753b4366ea Mon Sep 17 00:00:00 2001 From: Linze Shi Date: Wed, 16 Sep 2026 19:45:31 -0700 Subject: [PATCH] [Bugfix] Update ComfyUI serving-mode mock after helper rename (#7694) Signed-off-by: Linze-Shi --- tests/e2e/features/comfyui/test_comfyui_integration.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/e2e/features/comfyui/test_comfyui_integration.py b/tests/e2e/features/comfyui/test_comfyui_integration.py index 3c6f0492a..480381f0a 100644 --- a/tests/e2e/features/comfyui/test_comfyui_integration.py +++ b/tests/e2e/features/comfyui/test_comfyui_integration.py @@ -534,7 +534,7 @@ def mock_async_omni( # 1. The API layer uses its stage_list and stage_configs attributes # 2. Its __init__ method has slow side effects (model & config loading). # These cases use the non-duplex mock engine without loading model configs. - mocker.patch("vllm_omni.entrypoints.openai.api_server._is_duplex_model", return_value=False) + mocker.patch("vllm_omni.entrypoints.openai.api_server._should_serve_duplex", return_value=False) mock_async_omni_cls = mocker.patch("vllm_omni.entrypoints.openai.api_server.AsyncOmni") monkeypatch.setattr( "vllm_omni.entrypoints.openai.serving_chat.OmniOpenAIServingChat._preprocess_chat",