Commit 0961f5a4 authored by Woosuk Kwon's avatar Woosuk Kwon
Browse files

Add find method to sequence group

parent eb52db1b
......@@ -76,3 +76,9 @@ class SequenceGroup:
return len(self.seqs)
else:
return len([seq for seq in self.seqs if seq.status == status])
def find(self, seq_id: int) -> Sequence:
for seq in self.seqs:
if seq.seq_id == seq_id:
return seq
raise ValueError(f'Sequence {seq_id} not found.')
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