Skip to content

Commit

Permalink
fix: Append allowed_plugins in tb-gcp-uploader to default allowed plu…
Browse files Browse the repository at this point in the history
…gins

PiperOrigin-RevId: 629223881
  • Loading branch information
vertex-sdk-bot authored and Copybara-Service committed Apr 29, 2024
1 parent da23f04 commit aab9c3e
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion google/cloud/aiplatform/tensorboard/uploader_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -103,14 +103,22 @@ def main(argv):
experiment_name, FLAGS.experiment_display_name, project_id, region
)

plugins = uploader_constants.ALLOWED_PLUGINS
if FLAGS.allowed_plugins:
plugins += [
plugin
for plugin in FLAGS.allowed_plugins
if plugin not in uploader_constants.ALLOWED_PLUGINS
]

tb_uploader = uploader.TensorBoardUploader(
experiment_name=experiment_name,
experiment_display_name=experiment_display_name,
description=FLAGS.description,
tensorboard_resource_name=FLAGS.tensorboard_resource_name,
blob_storage_bucket=blob_storage_bucket,
blob_storage_folder=blob_storage_folder,
allowed_plugins=FLAGS.allowed_plugins,
allowed_plugins=plugins,
writer_client=api_client,
logdir=FLAGS.logdir,
one_shot=FLAGS.one_shot,
Expand Down

0 comments on commit aab9c3e

Please sign in to comment.