Skip to content
Snippets Groups Projects
Commit c3cf271d authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

fix

parent 2015dfd6
No related branches found
No related tags found
No related merge requests found
......@@ -13,6 +13,12 @@ from mhcflurry.class1_allele_specific import cv_and_train_command
from mhcflurry import downloads, predict
from mhcflurry.class1_allele_specific.train import HYPERPARAMETER_DEFAULTS
try:
import kubeface
KUBEFACE_INSTALLED = True
except ImportError:
KUBEFACE_INSTALLED = False
def test_small_run():
base_temp_dir = tempfile.mkdtemp()
......@@ -61,8 +67,9 @@ def test_small_run():
"--alleles", "HLA-A0201", "HLA-A0301",
"--verbose",
"--num-local-threads", "1",
"--storage-prefix", "/tmp/",
]
if KUBEFACE_INSTALLED:
args.extend(["--storage-prefix", "/tmp/"])
print("Running cv_and_train_command with args: %s " % str(args))
cv_and_train_command.run(args)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment