mirror of
https://github.com/microsoft/autogen.git
synced 2026-01-23 08:58:16 -05:00
catch import error
This commit is contained in:
@@ -651,7 +651,12 @@ class TestAutoML(unittest.TestCase):
|
||||
|
||||
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()
|
||||
|
||||
Reference in New Issue
Block a user