From 7809ec15ac7c9da87ee08f8e76b77ca8799beb59 Mon Sep 17 00:00:00 2001 From: Chi Wang Date: Tue, 19 Oct 2021 09:03:10 -0700 Subject: [PATCH] catch import error --- test/test_automl.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/test/test_automl.py b/test/test_automl.py index 85f39221f7..6714dd85a4 100644 --- a/test/test_automl.py +++ b/test/test_automl.py @@ -651,7 +651,12 @@ def test_parallel_classification(self): X, y = make_classification(1000, 10) automl = AutoML() - automl.fit(X, y, time_budget=10, task="classification", n_concurrent_trials=2) + try: + automl.fit( + X, y, time_budget=10, task="classification", n_concurrent_trials=2 + ) + except ImportError: + return def test_parallel_xgboost(self, hpo_method=None): automl_experiment = AutoML()