From 66e6eaebbe2ac4e9cd370afa32242ae9117beece Mon Sep 17 00:00:00 2001
From: A Vertex SDK engineer <vertex-sdk-bot@google.com>
Date: Wed, 31 May 2023 15:35:38 -0700
Subject: [PATCH] fix: fix the bug that start_upload_tb_log() doesn't
 continuously upload

PiperOrigin-RevId: 536842211
---
 google/cloud/aiplatform/tensorboard/uploader_tracker.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/google/cloud/aiplatform/tensorboard/uploader_tracker.py b/google/cloud/aiplatform/tensorboard/uploader_tracker.py
index 0fc4c1c042..daf0f6541d 100644
--- a/google/cloud/aiplatform/tensorboard/uploader_tracker.py
+++ b/google/cloud/aiplatform/tensorboard/uploader_tracker.py
@@ -254,7 +254,7 @@ def _create_uploader(
             allowed_plugins=uploader_constants.ALLOWED_PLUGINS,
             writer_client=api_client,
             logdir=logdir,
-            one_shot=True,
+            one_shot=one_shot,
             run_name_prefix=run_name_prefix,
             description=description,
             verbosity=verbosity,