chore: Pin system test ray cluster version as 2.4 until ready to update testing dependencies to ray 2.9

matthew29tang · copybara-github · commit e004e87ceccd · 2024-03-11T15:49:06.000-07:00
PiperOrigin-RevId: 614819864
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)

Original file line number	Diff line number	Diff line change
`@@ -43,6 +43,7 @@ def test_cluster_management(self):`
`43`	`43`	`head_node_type=head_node_type,`
`44`	`44`	`worker_node_types=worker_node_types,`
`45`	`45`	`cluster_name=f"ray-cluster-{timestamp}-test-cluster-management",`
	`46`	`+ ray_version="2.4",`
`46`	`47`	`)`
`47`	`48`
`48`	`49`	`cluster_details = vertex_ray.get_ray_cluster(cluster_resource_name)`