chore: Pin system test ray cluster version as 2.4 until ready to upda…

…te testing dependencies to ray 2.9 PiperOrigin-RevId: 614819864
googleapis · Mar 11, 2024 · e004e87 · e004e87
1 parent b91edf5
commit e004e87
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/tests/system/vertex_ray/test_cluster_management.py b/tests/system/vertex_ray/test_cluster_management.py
@@ -43,6 +43,7 @@ def test_cluster_management(self):
             head_node_type=head_node_type,
             worker_node_types=worker_node_types,
             cluster_name=f"ray-cluster-{timestamp}-test-cluster-management",
+            ray_version="2.4",
         )
 
         cluster_details = vertex_ray.get_ray_cluster(cluster_resource_name)

diff --git a/tests/system/vertex_ray/test_job_submission_dashboard.py b/tests/system/vertex_ray/test_job_submission_dashboard.py
@@ -26,7 +26,6 @@
 import tempfile
 
 RAY_VERSION = "2.4.0"
-CLUSTER_RAY_VERSION = "2_4"
 PROJECT_ID = "ucaip-sample-tests"
 
 

diff --git a/tests/system/vertex_ray/test_ray_data.py b/tests/system/vertex_ray/test_ray_data.py
@@ -26,7 +26,6 @@
 import tempfile
 
 RAY_VERSION = "2.4.0"
-CLUSTER_RAY_VERSION = "2_4"
 SDK_VERSION = aiplatform.__version__
 PROJECT_ID = "ucaip-sample-tests"
 
@@ -52,6 +51,7 @@ def test_ray_data(self):
             head_node_type=head_node_type,
             worker_node_types=worker_node_types,
             cluster_name=f"ray-cluster-{timestamp}-test-ray-data",
+            ray_version="2.4",
         )
 
         cluster_details = vertex_ray.get_ray_cluster(cluster_resource_name)