mirror of
https://github.com/microsoft/autogen.git
synced 2026-02-17 01:31:26 -05:00
"intermediate_results" TypeError: argument of type 'NoneType' is not iterable (#695)
* fix mlflow bug * bump version
This commit is contained in:
@@ -154,6 +154,19 @@ def test_mlflow():
|
||||
pass
|
||||
# subprocess.check_call([sys.executable, "-m", "pip", "uninstall", "mlflow"])
|
||||
|
||||
from sklearn.datasets import load_iris
|
||||
|
||||
with mlflow.start_run():
|
||||
automl = AutoML()
|
||||
automl_settings = {
|
||||
"time_budget": 2, # in seconds
|
||||
"metric": "accuracy",
|
||||
"task": "classification",
|
||||
"log_file_name": "iris.log",
|
||||
}
|
||||
X_train, y_train = load_iris(return_X_y=True)
|
||||
automl.fit(X_train=X_train, y_train=y_train, **automl_settings)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
test_automl(600)
|
||||
|
||||
Reference in New Issue
Block a user