#179 use var instead of magic numbers

MannLabs · Jul 2, 2024 · fc0174e · fc0174e
1 parent ca2ca32
commit fc0174e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/peptdeep/hla/hla_class1.py b/peptdeep/hla/hla_class1.py
@@ -166,7 +166,8 @@ def _prepare_train_data_df(self,
         precursor_df['nAA'] = precursor_df.sequence.str.len()
         precursor_df.drop(
             index=precursor_df[
-                (precursor_df.nAA<8)|(precursor_df.nAA>14)
+                (precursor_df.nAA<self.min_peptide_length)
+                |(precursor_df.nAA>self.max_peptide_length)
             ].index,
             inplace=True,
         )