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

PiperOrigin-RevId: 536842211
googleapis · May 31, 2023 · 66e6eae · 66e6eae
1 parent 162b2f2
commit 66e6eae
Showing 1 changed file with 1 addition and 1 deletion.
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,