Enhance OmniScheduler error handling with logging

- Added logging for exceptions when wrapping scheduled requests in OmniScheduler.
- Removed unnecessary blank lines for cleaner code structure.
This commit is contained in:
tzhouam committed 2025-10-24 12:00:36 +08:00
1 parent 87b3ae7e70
commit eb006f66e5
1 file changed
+2 -3
+2 -3
View File
@@ -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