diff --git a/mhcflurry/class1_affinity_predictor.py b/mhcflurry/class1_affinity_predictor.py index 85ecc1a1dc023b16f32810e2d988f3f7a0d1a257..c8f1d0e9a1ac5057e557401fe11c3b7d2057f7a3 100644 --- a/mhcflurry/class1_affinity_predictor.py +++ b/mhcflurry/class1_affinity_predictor.py @@ -702,7 +702,6 @@ class Class1AffinityPredictor(object): progress_print_interval=progress_print_interval) model_name = self.model_name("pan-class1", i) - self.class1_pan_allele_models.append(model) row = pandas.Series(collections.OrderedDict([ ("model_name", model_name), ("allele", "pan-class1"), @@ -711,6 +710,7 @@ class Class1AffinityPredictor(object): ])).to_frame().T self._manifest_df = pandas.concat( [self.manifest_df, row], ignore_index=True) + self.class1_pan_allele_models.append(model) if models_dir_for_save: self.save( models_dir_for_save, model_names_to_write=[model_name])