From ac17d876074f3fb51ab6c04beff0d3985df54633 Mon Sep 17 00:00:00 2001 From: A Vertex SDK engineer Date: Tue, 28 May 2024 09:07:43 -0700 Subject: [PATCH] fix: Create run_name when run_name_prefix is not specified for Tensorboard uploader. PiperOrigin-RevId: 637919611 --- .../tensorboard/plugins/tf_profiler/profile_uploader.py | 5 ++++- google/cloud/aiplatform/tensorboard/uploader_utils.py | 3 +++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/google/cloud/aiplatform/tensorboard/plugins/tf_profiler/profile_uploader.py b/google/cloud/aiplatform/tensorboard/plugins/tf_profiler/profile_uploader.py index 049864346b..70a3775ab5 100644 --- a/google/cloud/aiplatform/tensorboard/plugins/tf_profiler/profile_uploader.py +++ b/google/cloud/aiplatform/tensorboard/plugins/tf_profiler/profile_uploader.py @@ -176,7 +176,10 @@ def send_request(self, run_name: str): self._profile_dir(run_name) ) - tb_run = self._one_platform_resource_manager.get_run_resource_name(run_name) + tensorboard_run_name = run_name if run_name else "profile" + tb_run = self._one_platform_resource_manager.get_run_resource_name( + tensorboard_run_name + ) if run_name not in self._run_to_file_request_sender: self._run_to_file_request_sender[ diff --git a/google/cloud/aiplatform/tensorboard/uploader_utils.py b/google/cloud/aiplatform/tensorboard/uploader_utils.py index bb71c80ceb..807c58a412 100644 --- a/google/cloud/aiplatform/tensorboard/uploader_utils.py +++ b/google/cloud/aiplatform/tensorboard/uploader_utils.py @@ -23,6 +23,7 @@ import re import time from typing import Callable, Dict, Generator, List, Optional, Tuple +import uuid from absl import app from google.api_core import exceptions @@ -225,6 +226,8 @@ def _create_or_get_run_resource( project=project, location=location, run_name=run_name ) if not experiment_run: + if not run_name: + run_name = str(uuid.uuid4()) experiment_run = experiment_run_resource.ExperimentRun.create( project=project, location=location,