Skip to content

Commit ac17d87

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
fix: Create run_name when run_name_prefix is not specified for Tensorboard uploader.
PiperOrigin-RevId: 637919611
1 parent 94d838d commit ac17d87

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

google/cloud/aiplatform/tensorboard/plugins/tf_profiler/profile_uploader.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,10 @@ def send_request(self, run_name: str):
176176
self._profile_dir(run_name)
177177
)
178178

179-
tb_run = self._one_platform_resource_manager.get_run_resource_name(run_name)
179+
tensorboard_run_name = run_name if run_name else "profile"
180+
tb_run = self._one_platform_resource_manager.get_run_resource_name(
181+
tensorboard_run_name
182+
)
180183

181184
if run_name not in self._run_to_file_request_sender:
182185
self._run_to_file_request_sender[

google/cloud/aiplatform/tensorboard/uploader_utils.py

+3
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
import re
2424
import time
2525
from typing import Callable, Dict, Generator, List, Optional, Tuple
26+
import uuid
2627

2728
from absl import app
2829
from google.api_core import exceptions
@@ -225,6 +226,8 @@ def _create_or_get_run_resource(
225226
project=project, location=location, run_name=run_name
226227
)
227228
if not experiment_run:
229+
if not run_name:
230+
run_name = str(uuid.uuid4())
228231
experiment_run = experiment_run_resource.ExperimentRun.create(
229232
project=project,
230233
location=location,

0 commit comments

Comments
 (0)