From eb006f66e5ad96715b2e5690a90a6747087f806b Mon Sep 17 00:00:00 2001 From: tzhouam Date: Fri, 24 Oct 2025 12:00:36 +0800 Subject: [PATCH] Enhance OmniScheduler error handling with logging - Added logging for exceptions when wrapping scheduled requests in OmniScheduler. - Removed unnecessary blank lines for cleaner code structure. --- vllm_omni/core/sched/scheduler.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/vllm_omni/core/sched/scheduler.py b/vllm_omni/core/sched/scheduler.py index 5a021949d..c2ab49b6b 100644 --- a/vllm_omni/core/sched/scheduler.py +++ b/vllm_omni/core/sched/scheduler.py @@ -1,8 +1,7 @@ from __future__ import annotations from vllm.v1.core.sched.output import SchedulerOutput from vllm.v1.core.sched.scheduler import Scheduler as VLLMScheduler - - +from vllm.logger import init_logger class OmniScheduler(VLLMScheduler): """ @@ -44,6 +43,6 @@ class OmniScheduler(VLLMScheduler): scheduler_output.scheduled_new_reqs = new_list # type: ignore[assignment] except Exception: # If anything goes wrong, leave the original output unchanged - pass + init_logger(__name__).exception("Failed to wrap scheduled_new_reqs with OmniNewRequestData") return scheduler_output \ No newline at end of file