fix

mhcflurry/class1_neural_network.py

@@ -353,6 +353,13 @@ class Class1NeuralNetwork(object):
         result['prediction_cache'] = None
         return result
 
+    def __setstate__(self, state):
+        """
+        Deserialize. For pickle support.
+        """
+        self.__dict__.update(state)
+        self.prediction_cache = weakref.WeakKeyDictionary()
+
     def peptides_to_network_input(self, peptides):
         """
         Encode peptides to the fixed-length encoding expected by the neural

mhcflurry/select_allele_specific_models_command.py

@@ -153,6 +153,8 @@ def run(argv=sys.argv[1:]):
 
             df = df.loc[~df._excluded]
             print("Reduced data to: %s" % (str(df.shape)))
+
+        metadata_dfs["model_selection_data"] = df
     else:
         df = None
 
@@ -195,8 +197,7 @@ def run(argv=sys.argv[1:]):
         print("Attempting to create directory: %s" % args.out_models_dir)
         os.mkdir(args.out_models_dir)
         print("Done.")
-
-    metadata_dfs["model_selection_data"] = df
+    
     result_predictor = Class1AffinityPredictor(metadata_dataframes=metadata_dfs)
 
     worker_pool = worker_pool_with_gpu_assignments_from_args(args)