fix: fix the bug that start_upload_tb_log() doesn't continuously upload

vertex-sdk-bot · copybara-github · commit 66e6eaebbe2a · 2023-05-31T15:36:58.000-07:00
PiperOrigin-RevId: 536842211
diff --git 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,
