fix: use https to connect through Ray dashboard

PiperOrigin-RevId: 601255907
googleapis · Jan 24, 2024 · 7af3e67 · 7af3e67
1 parent 262a36b
commit 7af3e67
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/google/cloud/aiplatform/preview/vertex_ray/dashboard_sdk.py b/google/cloud/aiplatform/preview/vertex_ray/dashboard_sdk.py
@@ -56,6 +56,7 @@ def get_job_submission_client_cluster_info(
             }
         return oss_dashboard_sdk.get_job_submission_client_cluster_info(
             address=address,
+            _use_tls=True,
             *args,
             **kwargs,
         )

diff --git a/tests/unit/vertex_ray/test_dashboard_sdk.py b/tests/unit/vertex_ray/test_dashboard_sdk.py
@@ -109,5 +109,6 @@ def test_job_submission_client_cluster_info_with_dashboard_address(
         get_bearer_token_mock.assert_called_once_with()
         ray_get_job_submission_client_cluster_info_mock.assert_called_once_with(
             address=tc.ClusterConstants._TEST_VERTEX_RAY_DASHBOARD_ADDRESS,
+            _use_tls=True,
             headers=tc.ClusterConstants._TEST_HEADERS,
         )