Skip to content

GitLab

Commit b39f149a authored by Woosuk Kwon's avatar Woosuk Kwon
Browse files

Add is_finished

parent ef6098ec
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
cacheflow/sequence.py
View file @ b39f149a
...@@ -86,6 +86,9 @@ class SequenceGroup: ...@@ -86,6 +86,9 @@ class SequenceGroup:
return seq return seq
raise ValueError(f'Sequence {seq_id} not found.') 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: def __repr__(self) -> str:
return (f'SequenceGroup(group_id={self.group_id}, ' return (f'SequenceGroup(group_id={self.group_id}, '
f'num_seqs={len(self.seqs)})') f'num_seqs={len(self.seqs)})')
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment