fix: use https to connect through Ray dashboard

yinghsienwu · copybara-github · commit 7af3e67654e0 · 2024-01-24T15:41:41.000-08:00
PiperOrigin-RevId: 601255907
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,
         )

Original file line number	Diff line number	Diff line change
`@@ -56,6 +56,7 @@ def get_job_submission_client_cluster_info(`
`56`	`56`	`}`
`57`	`57`	`return oss_dashboard_sdk.get_job_submission_client_cluster_info(`
`58`	`58`	`address=address,`
	`59`	`+ _use_tls=True,`
`59`	`60`	`*args,`
`60`	`61`	`**kwargs,`
`61`	`62`	`)`
Original file line number	Diff line number	Diff line change
`@@ -109,5 +109,6 @@ def test_job_submission_client_cluster_info_with_dashboard_address(`
`109`	`109`	`get_bearer_token_mock.assert_called_once_with()`
`110`	`110`	`ray_get_job_submission_client_cluster_info_mock.assert_called_once_with(`
`111`	`111`	`address=tc.ClusterConstants._TEST_VERTEX_RAY_DASHBOARD_ADDRESS,`
	`112`	`+ _use_tls=True,`
`112`	`113`	`headers=tc.ClusterConstants._TEST_HEADERS,`
`113`	`114`	`)`