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:
1 parent
87b3ae7e70
commit
eb006f66e5
1 file changed
+2
-3
@@ -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
|
||||
Reference in new issue
Block a user