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 <[email protected]>
microsoft · Feb 9, 2022 · b4d3124 · b4d3124
1 parent a4970d2
commit b4d3124
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -66,8 +66,7 @@
         "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
@@ -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
     )