wangtao
/
autogen
mirror of https://github.com/microsoft/autogen.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

catch import error

This commit is contained in:
Chi Wang 2021-10-19 09:03:10 -07:00
parent 29fac8807b
commit 7809ec15ac
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
test/test_automl.py
View File

@ -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()