feat: Populate Ray Cluster dashboard_address from proto field

PiperOrigin-RevId: 582853171
googleapis · Nov 16, 2023 · dd4b852 · dd4b852
1 parent 1f81cf2
commit dd4b852
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/google/cloud/aiplatform/preview/vertex_ray/util/_gapic_utils.py b/google/cloud/aiplatform/preview/vertex_ray/util/_gapic_utils.py
@@ -138,11 +138,15 @@ def persistent_resource_to_cluster(
     Returns:
         Cluster.
     """
+    dashboard_address = persistent_resource.resource_runtime.access_uris.get(
+        "RAY_DASHBOARD_URI"
+    )
     cluster = Cluster(
         cluster_resource_name=persistent_resource.name,
         network=persistent_resource.network,
         state=persistent_resource.state.name,
         labels=persistent_resource.labels,
+        dashboard_address=dashboard_address,
     )
     if not persistent_resource.resource_runtime_spec.ray_spec:
         # skip PersistentResource without RaySpec