From 6aa196c8dc977abebbb1a721279944d21f9ca174 Mon Sep 17 00:00:00 2001 From: Nick Hill Date: Tue, 25 Mar 2025 14:21:36 -0700 Subject: [PATCH] [V1][Minor] Use `SchedulerInterface` type for engine scheduler field (#15499) Signed-off-by: Nick Hill --- vllm/v1/engine/core.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vllm/v1/engine/core.py b/vllm/v1/engine/core.py index f4bb4583bea48..42511777feebb 100644 --- a/vllm/v1/engine/core.py +++ b/vllm/v1/engine/core.py @@ -23,6 +23,7 @@ from vllm.utils import (get_exception_traceback, resolve_obj_by_qualname, zmq_socket_ctx) from vllm.v1.core.kv_cache_utils import (get_kv_cache_config, unify_kv_cache_configs) +from vllm.v1.core.sched.interface import SchedulerInterface from vllm.v1.core.sched.output import SchedulerOutput from vllm.v1.core.sched.scheduler import Scheduler as V1Scheduler from vllm.v1.engine import (EngineCoreOutputs, EngineCoreRequest, @@ -84,7 +85,7 @@ class EngineCore: "compatibility may not be maintained.", vllm_config.scheduler_config.scheduler_cls) - self.scheduler = Scheduler( + self.scheduler: SchedulerInterface = Scheduler( scheduler_config=vllm_config.scheduler_config, model_config=vllm_config.model_config, cache_config=vllm_config.cache_config,