In train_allele_specific_models_command.py, serial run is when num jobs = 0 (not 1).

Fixes #149

mhcflurry/train_allele_specific_models_command.py

@@ -193,7 +193,7 @@ def run(argv=sys.argv[1:]):
         metadata_dataframes={
             'train_data': df,
         })
-    serial_run = args.num_jobs == 1
+    serial_run = args.num_jobs == 0
 
     work_items = []
     for (h, hyperparameters) in enumerate(hyperparameters_lst):

test/test_train_and_related_commands.py

@@ -164,12 +164,11 @@ def run_and_check_with_model_selection(n_jobs=1):
     shutil.rmtree(models_dir1)
 
 
-if os.environ.get("KERAS_BACKEND") != "theano":
-    def test_run_parallel():
-        run_and_check(n_jobs=2)
-        run_and_check_with_model_selection(n_jobs=2)
+def test_run_parallel():
+    run_and_check(n_jobs=2)
+    run_and_check_with_model_selection(n_jobs=2)
 
 
 def test_run_serial():
-    run_and_check(n_jobs=1)
-    run_and_check_with_model_selection(n_jobs=1)
\ No newline at end of file
+    run_and_check(n_jobs=0)
+    run_and_check_with_model_selection(n_jobs=0)
\ No newline at end of file