cast web_access_uris to dict type

googleapis · Oct 20, 2021 · 71f5813 · 71f5813
1 parent 5e44f24
commit 71f5813
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/google/cloud/aiplatform/jobs.py b/google/cloud/aiplatform/jobs.py
@@ -1108,7 +1108,7 @@ def _get_web_access_uris(self) -> Dict[str, str]:
             (Dict[str, str]):
                 Web access uris of the custom job.
         """
-        return self._gca_resource.web_access_uris
+        return dict(self._gca_resource.web_access_uris)
 
     def _log_web_access_uris(self):
         """Helper method to log the web access uris of the custom job"""
@@ -1678,9 +1678,9 @@ def _get_web_access_uris(self) -> Dict[str, Dict[str, str]]:
             (Dict[str, Dict[str, str]]):
                 Web access uris of the hyperparameter job.
         """
-        web_access_uris = {}
+        web_access_uris = dict()
         for trial in self.trials:
-            web_access_uris[trial.id] = web_access_uris.get(trial.id, {})
+            web_access_uris[trial.id] = web_access_uris.get(trial.id, dict())
             for worker, uri in trial.web_access_uris.items():
                 web_access_uris[trial.id][worker] = uri
         return web_access_uris

diff --git a/google/cloud/aiplatform/training_jobs.py b/google/cloud/aiplatform/training_jobs.py
@@ -1457,7 +1457,7 @@ def web_access_uris(self) -> Dict[str, str]:
             (Dict[str, str]):
                 Web access uris of the backing custom job.
         """
-        web_access_uris = {}
+        web_access_uris = dict()
         if (
             self._gca_resource.training_task_metadata
             and self._gca_resource.training_task_metadata.get("backingCustomJob")
@@ -1467,7 +1467,7 @@ def web_access_uris(self) -> Dict[str, str]:
             )
             custom_job = jobs.CustomJob.get(resource_name=custom_job_resource_name)
 
-            web_access_uris = custom_job.web_access_uris
+            web_access_uris = dict(custom_job.web_access_uris)
 
         return web_access_uris