fix: Ray - Fixed exception when using Ray 2.4

PiperOrigin-RevId: 630196444
googleapis · May 2, 2024 · 2661f52 · 2661f52
1 parent eb05ac4
commit 2661f52
Showing 1 changed file with 1 addition and 1 deletion.
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: