From 40be2fafbbcf968713de9b3ddd4e76495b07ea1d Mon Sep 17 00:00:00 2001 From: Googler Date: Fri, 8 Mar 2024 17:06:19 -0800 Subject: [PATCH] feat(components): internal PiperOrigin-RevId: 614085882 --- .../preview/automl/vision/data_converter.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/components/google-cloud/google_cloud_pipeline_components/preview/automl/vision/data_converter.py b/components/google-cloud/google_cloud_pipeline_components/preview/automl/vision/data_converter.py index 74156c979df4..702a7a8c6877 100644 --- a/components/google-cloud/google_cloud_pipeline_components/preview/automl/vision/data_converter.py +++ b/components/google-cloud/google_cloud_pipeline_components/preview/automl/vision/data_converter.py @@ -22,6 +22,7 @@ from kfp import dsl +# pylint: disable=singleton-comparison # pylint: disable=g-doc-args @dsl.container_component def data_converter( @@ -31,6 +32,7 @@ def data_converter( objective: str, output_dir: dsl.Output[dsl.Artifact], gcp_resources: dsl.OutputPath(str), + enable_input_validation: bool = True, location: str = 'us-central1', timeout: str = '604800s', service_account: Optional[str] = None, @@ -75,7 +77,7 @@ def data_converter( 'image_uri': 'us-docker.pkg.dev/vertex-ai/vertex-vision-model-garden-dockers/data-converter', 'args': [ '--enable_input_validation', - 'true', + enable_input_validation, '--input_file_path', input_file_path, '--input_file_type',