diff --git a/mhcflurry/encodable_sequences.py b/mhcflurry/encodable_sequences.py index efe29c1477cd693792ede926a0751c752dc25a7a..a27aaf4459a9819fcee59d7ffe87d0809ed7b88b 100644 --- a/mhcflurry/encodable_sequences.py +++ b/mhcflurry/encodable_sequences.py @@ -37,8 +37,8 @@ class EncodableSequences(object): raise ValueError("Sequence of strings is required") self.sequences_df = pandas.DataFrame({ "sequence": numpy.array(sequences), - "sequence_length": numpy.array(sequences), }) + self.sequences_df["sequence_length"] = self.sequences.str.len() self.min_length = self.sequences_df.sequence_length.min() self.max_length = self.sequences_df.sequence_length.max() self.sequences = self.sequences_df.sequence.values