Add find method to sequence group

This commit is contained in:
Woosuk Kwon 2023-02-13 02:39:12 +00:00
parent eb52db1bea
commit 0961f5a49a

View File

@ -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.')