catch import error

microsoft · Oct 19, 2021 · 7809ec1 · 7809ec1
1 parent 29fac88
commit 7809ec1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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()