diff --git a/mhcflurry/class1_affinity_prediction/class1_affinity_predictor.py b/mhcflurry/class1_affinity_prediction/class1_affinity_predictor.py index 21a92e72e1817667d5a5981946befa4240cb0c0f..d0e84024372ee97b7795bb590fc42b92a434b8f7 100644 --- a/mhcflurry/class1_affinity_prediction/class1_affinity_predictor.py +++ b/mhcflurry/class1_affinity_prediction/class1_affinity_predictor.py @@ -558,8 +558,8 @@ class Class1AffinityPredictor(object): df["allele"] = alleles df["result"] = numpy.nan for (allele, sub_df) in df.groupby("allele"): - df.loc[sub_df.index, "result"] = self.percentile_ranks\ - (sub_df.affinity, allele=allele) + df.loc[sub_df.index, "result"] = self.percentile_ranks( + sub_df.affinity, allele=allele) assert not df.result.isnull().any() return df.result.values