From b4d312412a6b11ab7d799f670eeeedcca9348f6f Mon Sep 17 00:00:00 2001 From: Chi Wang Date: Wed, 9 Feb 2022 15:04:29 -0800 Subject: [PATCH] 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 --- setup.py | 3 +-- test/nlp/test_autohf_regression.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index d371ed5bd..907f1fe50 100644 --- a/setup.py +++ b/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", diff --git a/test/nlp/test_autohf_regression.py b/test/nlp/test_autohf_regression.py index 1854f7214..762091b01 100644 --- a/test/nlp/test_autohf_regression.py +++ b/test/nlp/test_autohf_regression.py @@ -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 )