chore(components): Fix bug due to protobuf library being upgraded by …

…pinning protobuf version Signed-off-by: Googler <[email protected]> PiperOrigin-RevId: 652610202
kubeflow · Jul 15, 2024 · 9cb5913 · 9cb5913
1 parent 4930bee
commit 9cb5913
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git 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/...ud_pipeline_components/_implementation/starry_net/upload_decomposition_plots/component.py b/...ud_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,