From 70c28425952f1de01bf715d8a9d7f2341ac3fbe9 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Wed, 23 May 2018 13:24:24 -0400
Subject: [PATCH] fix

---
 mhcflurry/class1_affinity_predictor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mhcflurry/class1_affinity_predictor.py b/mhcflurry/class1_affinity_predictor.py
index 85ecc1a1..c8f1d0e9 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])
-- 
GitLab