From b39f149a0853c6b6d48716f21fbb7b97828d17f4 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Fri, 24 Feb 2023 11:44:21 +0000 Subject: [PATCH] Add is_finished --- cacheflow/sequence.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/cacheflow/sequence.py b/cacheflow/sequence.py index 9f140f0b..364f3b2e 100644 --- a/cacheflow/sequence.py +++ b/cacheflow/sequence.py @@ -86,6 +86,9 @@ class SequenceGroup: return seq raise ValueError(f'Sequence {seq_id} not found.') + def is_finished(self) -> bool: + return all(seq.status == SequenceStatus.FINISHED for seq in self.seqs) + def __repr__(self) -> str: return (f'SequenceGroup(group_id={self.group_id}, ' f'num_seqs={len(self.seqs)})')