diff --git a/components/google-cloud/RELEASE.md b/components/google-cloud/RELEASE.md index 0a8b52659ee..15e314f2166 100644 --- a/components/google-cloud/RELEASE.md +++ b/components/google-cloud/RELEASE.md @@ -1,5 +1,6 @@ ## Upcoming release * Updated the Starry Net pipeline's template gallery description, and added dataprep_nan_threshold and dataprep_zero_threshold args to the Starry Net pipeline. +* Fix bug in Starry Net's upload decomposition plot step due to protobuf upgrade, by pinning protobuf library to 3.20.*. * Add support for running tasks on a `PersistentResource` (see [CustomJobSpec](https://cloud.google.com/vertex-ai/docs/reference/rest/v1beta1/CustomJobSpec)) via `persistent_resource_id` parameter on `v1.custom_job.CustomTrainingJobOp` and `v1.custom_job.create_custom_training_job_from_component` * Bump image for Structured Data pipelines. diff --git a/components/google-cloud/google_cloud_pipeline_components/_implementation/starry_net/upload_decomposition_plots/component.py b/components/google-cloud/google_cloud_pipeline_components/_implementation/starry_net/upload_decomposition_plots/component.py index fe958fa2bb7..43ce7d0e9f8 100644 --- a/components/google-cloud/google_cloud_pipeline_components/_implementation/starry_net/upload_decomposition_plots/component.py +++ b/components/google-cloud/google_cloud_pipeline_components/_implementation/starry_net/upload_decomposition_plots/component.py @@ -16,7 +16,12 @@ from kfp import dsl -@dsl.component(packages_to_install=['google-cloud-aiplatform[tensorboard]']) +@dsl.component( + packages_to_install=[ + 'google-cloud-aiplatform[tensorboard]', + 'protobuf==3.20.*', + ] +) def upload_decomposition_plots( project: str, location: str,