[BugFix] Fix abort_seq_group (#2463)

This commit is contained in:
ljss
2024-01-19 07:10:42 +08:00
committed by GitHub
parent 7e1081139d
commit d2a68364c4

View File

@@ -108,7 +108,7 @@ class Scheduler:
request_id = (request_id, )
request_ids = set(request_id)
for state_queue in [self.waiting, self.running, self.swapped]:
aborted_groups = []
aborted_groups: List[SequenceGroup] = []
for seq_group in state_queue:
if not request_ids:
# Using 'break' here may add two extra iterations,
@@ -121,7 +121,7 @@ class Scheduler:
for aborted_group in aborted_groups:
# Remove the sequence group from the state queue.
state_queue.remove(aborted_group)
for seq in seq_group.get_seqs():
for seq in aborted_group.get_seqs():
if seq.is_finished():
continue
seq.status = SequenceStatus.FINISHED_ABORTED