mirror of
https://github.com/microsoft/FLAML.git
synced 2026-02-09 02:09:16 +08:00
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:
3
setup.py
3
setup.py
@@ -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",
|
||||
|
||||
@@ -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
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user