fix

mhcflurry/train_pan_allele_models_command.py

@@ -32,8 +32,6 @@ from .allele_encoding import AlleleEncoding
 from .encodable_sequences import EncodableSequences
 from .regression_target import to_ic50, from_ic50
 
-os.environ["CUDA_VISIBLE_DEVICES"] = ""
-
 
 # To avoid pickling large matrices to send to child processes when running in
 # parallel, we use this global variable as a place to store data. Data that is

test/test_train_and_related_commands.py

@@ -14,6 +14,8 @@ from numpy.testing import assert_array_less, assert_equal
 from mhcflurry import Class1AffinityPredictor
 from mhcflurry.downloads import get_path
 
+os.environ["CUDA_VISIBLE_DEVICES"] = ""
+
 HYPERPARAMETERS = [
     {
         "n_models": 2,

test/test_train_pan_allele_models_command.py

@@ -18,6 +18,8 @@ from mhcflurry import Class1AffinityPredictor,Class1NeuralNetwork
 from mhcflurry.allele_encoding import AlleleEncoding
 from mhcflurry.downloads import get_path
 
+os.environ["CUDA_VISIBLE_DEVICES"] = ""
+
 
 HYPERPARAMETERS_LIST = [
 {