fixes

mhcflurry/class1_allele_specific/cross_validation.py

@@ -192,4 +192,4 @@ def cross_validation_folds(
             for (result_fold, imputation_result) in zip(
                 result_folds, imputation_results)
         ]
-    return result_fold
+    return result_folds

mhcflurry/parallelism.py

@@ -14,6 +14,7 @@ class ParallelBackend(object):
         self.module = module
         self.verbose = verbose
 
+
 class KubefaceParallelBackend(ParallelBackend):
     """
     ParallelBackend that uses kubeface

test/test_class1_allele_specific_cv_and_train_command.py

@@ -61,6 +61,7 @@ def test_small_run():
         "--alleles", "HLA-A0201", "HLA-A0301",
         "--verbose",
         "--num-local-threads", "1",
+        "--storage-prefix", "/tmp/",
     ]
     print("Running cv_and_train_command with args: %s " % str(args))
 

test/test_cross_validation.py

@@ -77,7 +77,7 @@ def test_cross_validation_with_imputation():
         n_imputations=2, n_burn_in=1, n_nearest_columns=25)
     train_data = (
         mhcflurry.dataset.Dataset.from_csv(
-            get_path("data_kim2014" , "bdata.2009.mhci.public.1.txt"))
+            get_path("data_kim2014", "bdata.2009.mhci.public.1.txt"))
         .get_alleles(["HLA-A0201", "HLA-A0202", "HLA-A0301"]))
 
     folds = cross_validation_folds(