Skip to content

Commit aab9c3e

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
fix: Append allowed_plugins in tb-gcp-uploader to default allowed plugins
PiperOrigin-RevId: 629223881
1 parent da23f04 commit aab9c3e

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

google/cloud/aiplatform/tensorboard/uploader_main.py

+9-1
Original file line numberDiff line numberDiff line change
@@ -103,14 +103,22 @@ def main(argv):
103103
experiment_name, FLAGS.experiment_display_name, project_id, region
104104
)
105105

106+
plugins = uploader_constants.ALLOWED_PLUGINS
107+
if FLAGS.allowed_plugins:
108+
plugins += [
109+
plugin
110+
for plugin in FLAGS.allowed_plugins
111+
if plugin not in uploader_constants.ALLOWED_PLUGINS
112+
]
113+
106114
tb_uploader = uploader.TensorBoardUploader(
107115
experiment_name=experiment_name,
108116
experiment_display_name=experiment_display_name,
109117
description=FLAGS.description,
110118
tensorboard_resource_name=FLAGS.tensorboard_resource_name,
111119
blob_storage_bucket=blob_storage_bucket,
112120
blob_storage_folder=blob_storage_folder,
113-
allowed_plugins=FLAGS.allowed_plugins,
121+
allowed_plugins=plugins,
114122
writer_client=api_client,
115123
logdir=FLAGS.logdir,
116124
one_shot=FLAGS.one_shot,

0 commit comments

Comments
 (0)