From ada8a47b12015f5c04586616c9ff19b1ea040264 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Thu, 13 Mar 2025 18:10:05 -0700 Subject: [PATCH] minor Signed-off-by: Woosuk Kwon --- vllm/v1/core/sched/simple_scheduler.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/vllm/v1/core/sched/simple_scheduler.py b/vllm/v1/core/sched/simple_scheduler.py index 27e0968f53bad..561b4134e5870 100644 --- a/vllm/v1/core/sched/simple_scheduler.py +++ b/vllm/v1/core/sched/simple_scheduler.py @@ -373,8 +373,7 @@ class SimpleScheduler(SchedulerInterface): if request.status == RequestStatus.RUNNING: self.running.remove(request) - if request.request_id in self.scheduled_req_ids: - self.scheduled_req_ids.remove(request.request_id) + self.scheduled_req_ids.discard(request.request_id) else: self.waiting.remove(request) request.status = finished_status