fix test_class1_allele_specific_cv_and_train_comman

test/test_class1_allele_specific_cv_and_train_command.py

@@ -10,7 +10,7 @@ from os import mkdir, environ
 import pandas
 
 from mhcflurry.class1_allele_specific import cv_and_train_command
-from mhcflurry import downloads, predict
+from mhcflurry import downloads, predict, class1_allele_specific
 from mhcflurry.class1_allele_specific.train import HYPERPARAMETER_DEFAULTS
 
 try:
@@ -113,7 +113,11 @@ def verify_trained_models(base_temp_dir):
             ("HLA-A0201", "LLGLPAAEY", False),
         ], columns=("allele", "peptide", "binder"))
 
-        data["prediction"] = predict(data.allele, data.peptide).Prediction
+        data["prediction"] = predict(
+            data.allele,
+            data.peptide,
+            predictor=class1_allele_specific.get_downloaded_predictor()
+        ).Prediction
         print(data)
         mean_binder = data.ix[data.binder].prediction.mean()
         mean_nonbinder = data.ix[~data.binder].prediction.mean()