Skip to content

Commit 6aa196c

Browse files
authored
[V1][Minor] Use SchedulerInterface type for engine scheduler field (#15499)
Signed-off-by: Nick Hill <nhill@redhat.com>
1 parent a0dd7dc commit 6aa196c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vllm/v1/engine/core.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
zmq_socket_ctx)
2424
from vllm.v1.core.kv_cache_utils import (get_kv_cache_config,
2525
unify_kv_cache_configs)
26+
from vllm.v1.core.sched.interface import SchedulerInterface
2627
from vllm.v1.core.sched.output import SchedulerOutput
2728
from vllm.v1.core.sched.scheduler import Scheduler as V1Scheduler
2829
from vllm.v1.engine import (EngineCoreOutputs, EngineCoreRequest,
@@ -84,7 +85,7 @@ def __init__(
8485
"compatibility may not be maintained.",
8586
vllm_config.scheduler_config.scheduler_cls)
8687

87-
self.scheduler = Scheduler(
88+
self.scheduler: SchedulerInterface = Scheduler(
8889
scheduler_config=vllm_config.scheduler_config,
8990
model_config=vllm_config.model_config,
9091
cache_config=vllm_config.cache_config,

0 commit comments

Comments
 (0)