From f61dc8072fbbef76430291ef62d79b7dd265b0e3 Mon Sep 17 00:00:00 2001
From: Light Lin <lxrite@gmail.com>
Date: Fri, 20 Oct 2023 23:50:47 +0800
Subject: [PATCH] Fix type hints (#1427)

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

diff --git a/vllm/core/scheduler.py b/vllm/core/scheduler.py
index 516f23d2b2f8d..0c98c063c8694 100644
--- a/vllm/core/scheduler.py
+++ b/vllm/core/scheduler.py
@@ -275,7 +275,7 @@ class Scheduler:
         # Create input data structures.
         seq_group_metadata_list: List[SequenceGroupMetadata] = []
         for seq_group in scheduler_outputs.scheduled_seq_groups:
-            seq_data: Dict[int, List[SequenceData]] = {}
+            seq_data: Dict[int, SequenceData] = {}
             block_tables: Dict[int, List[int]] = {}
             for seq in seq_group.get_seqs(status=SequenceStatus.RUNNING):
                 seq_id = seq.seq_id