bump ray version to 1.10 (#450)

* bump ray version to 1.10

* init ray in test

* Update setup.py to include hotfixes

Co-authored-by: Antoni Baum <antoni.baum@protonmail.com>
This commit is contained in:
Chi Wang
2022-02-09 15:04:29 -08:00
committed by GitHub
parent a4970d279c
commit b4d312412a
2 changed files with 2 additions and 3 deletions

View File

@@ -66,8 +66,7 @@ setuptools.setup(
"catboost": ["catboost>=0.26"],
"blendsearch": ["optuna==2.8.0"],
"ray": [
"ray[tune]==1.6.0",
"pyyaml<5.3.1",
"ray[tune]~=1.10",
],
"azureml": [
"azureml-mlflow",

View File

@@ -76,7 +76,7 @@ def test_regression():
}
ray.shutdown()
ray.init()
automl.fit(
X_train=X_train, y_train=y_train, X_val=X_val, y_val=y_val, **automl_settings
)