fix: Ray - Fixed exception when using Ray 2.4

Ark-kun · copybara-github · commit 2661f52fd081 · 2024-05-02T15:30:05.000-07:00
PiperOrigin-RevId: 630196444
diff --git a/google/cloud/aiplatform/preview/vertex_ray/cluster_init.py b/google/cloud/aiplatform/preview/vertex_ray/cluster_init.py
@@ -130,7 +130,7 @@ def create_ray_cluster(
         )
     if ray_version == "2.4":
         warnings.warn(
-            _gapic_utils._V2_4_WARNING_MESSAGE, DeprecationWarning, stacklevel=2
+            _validation_utils._V2_4_WARNING_MESSAGE, DeprecationWarning, stacklevel=2
         )
     local_ray_verion = _validation_utils.get_local_ray_version()
     if ray_version != local_ray_verion:

Original file line number	Diff line number	Diff line change
`@@ -130,7 +130,7 @@ def create_ray_cluster(`
`130`	`130`	`)`
`131`	`131`	`if ray_version == "2.4":`
`132`	`132`	`warnings.warn(`
`133`		`- _gapic_utils._V2_4_WARNING_MESSAGE, DeprecationWarning, stacklevel=2`
	`133`	`+ _validation_utils._V2_4_WARNING_MESSAGE, DeprecationWarning, stacklevel=2`
`134`	`134`	`)`
`135`	`135`	`local_ray_verion = _validation_utils.get_local_ray_version()`
`136`	`136`	`if ray_version != local_ray_verion:`