From 800024de84f8ded2c5284edbb29700a8a7398325 Mon Sep 17 00:00:00 2001 From: gcanlin Date: Sun, 12 Apr 2026 13:46:25 +0000 Subject: [PATCH] [Bugfix] Fix UT for the missing of log_stats in Engine Signed-off-by: gcanlin --- tests/engine/test_async_omni_engine_stage_init.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/engine/test_async_omni_engine_stage_init.py b/tests/engine/test_async_omni_engine_stage_init.py index 24d2bf0cf9..f397307936 100644 --- a/tests/engine/test_async_omni_engine_stage_init.py +++ b/tests/engine/test_async_omni_engine_stage_init.py @@ -98,6 +98,7 @@ def test_initialize_stages_passes_stage_init_timeout_to_diffusion_handshake(monk from vllm_omni.platforms import current_omni_platform engine = object.__new__(AsyncOmniEngine) + engine.log_stats = False engine.model = "dummy-model" engine.config_path = "dummy-config" engine.num_stages = 1 @@ -178,6 +179,7 @@ def test_launch_llm_stage_passes_stage_init_timeout_to_complete_stage_handshake( from vllm_omni.platforms import current_omni_platform engine = object.__new__(AsyncOmniEngine) + engine.log_stats = False engine.model = "dummy-model" engine.single_stage_mode = False engine._omni_master_server = None