From c3cf271d122c04f1b6459ce47434613c7ba7ef8d Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Thu, 9 Feb 2017 14:22:09 -0500
Subject: [PATCH] fix
---
test/test_class1_allele_specific_cv_and_train_command.py | 9 ++++++++-
1 file changed, 8 insertions(+), 1 deletion(-)
diff --git a/test/test_class1_allele_specific_cv_and_train_command.py b/test/test_class1_allele_specific_cv_and_train_command.py
index 9f0c193b..99d1a991 100644
--- a/test/test_class1_allele_specific_cv_and_train_command.py
+++ b/test/test_class1_allele_specific_cv_and_train_command.py
@@ -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)
--
GitLab