Add is_finished

b39f149a · Woosuk Kwon · ef6098ec · b39f149a
Commit b39f149a authored Feb 24, 2023 by Woosuk Kwon
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

cacheflow/sequence.py cacheflow/sequence.py +3 -0

No files found.
--- 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)})')