From 98bbbefb68b0f7fa3edc50704143075d37cb1c31 Mon Sep 17 00:00:00 2001
From: MARD1NO <359521840@qq.com>
Date: Sun, 15 Jun 2025 10:15:05 +0800
Subject: [PATCH] schedule return bool args

---
 nanovllm/engine/scheduler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nanovllm/engine/scheduler.py b/nanovllm/engine/scheduler.py
index 6d1eb53..cbf6b65 100644
--- a/nanovllm/engine/scheduler.py
+++ b/nanovllm/engine/scheduler.py
@@ -23,7 +23,7 @@ class Scheduler:
     def add(self, seq: Sequence):
         self.waiting.append(seq)
 
-    def schedule(self) -> tuple[list[Sequence], SequenceStatus]:
+    def schedule(self) -> tuple[list[Sequence], bool]:
         # prefill
         scheduled_seqs = []
         num_seqs = 0