upgrade to ray 2.7.0

Signed-off-by: Kelly A <[email protected]>
caikit · Sep 29, 2023 · 59cc2b3 · 59cc2b3
1 parent 1cbe2cf
commit 59cc2b3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,8 +15,7 @@ classifiers=[
 ]
 dependencies = [
     "caikit>=0.15.0,<0.21.0",
-    "ray[default]==2.6.3",
-    "pydantic<2"
+    "ray[default]==2.7.0"
 ]
 
 [project.scripts]

diff --git a/tests/test_ray_backend.py b/tests/test_ray_backend.py
@@ -22,6 +22,7 @@
 # Third Party
 # Third party
 from ray import job_submission
+from ray.exceptions import RaySystemError
 from sample_lib.modules.sample_task.sample_implementation import SampleModule
 import pytest
 
@@ -190,7 +191,7 @@ def test_stop_cluster(mock_ray_cluster):
     config = {"connection": {"address": mock_ray_cluster.address}}
     ray_be = RayBackend(config)
     mock_ray_cluster.shutdown()
-    with pytest.raises(ConnectionError):
+    with pytest.raises(RaySystemError):
         client = ray_be.get_client()