update notebook and test

This commit is contained in:
Qingyun Wu 2022-03-30 22:01:18 +00:00 committed by Chi Wang
parent 9128c8811a
commit 2cdc08a75a
2 changed files with 586 additions and 415 deletions

File diff suppressed because one or more lines are too long

View File

@ -7,7 +7,7 @@ def test_automl(budget=5, dataset_format="dataframe", hpo_method=None):
from flaml.data import load_openml_dataset
import urllib3
performance_check_budget = 240
performance_check_budget = 600
if (
sys.platform == "darwin"
and budget < performance_check_budget
@ -141,4 +141,4 @@ def test_mlflow():
if __name__ == "__main__":
test_automl(240)
test_automl(600)