Skip to content

Commit a4f01b7

Browse files
author
Googler
committed
feat(components): internal
PiperOrigin-RevId: 627804544
1 parent ae530de commit a4f01b7

File tree

1 file changed

+3
-1
lines changed
  • components/google-cloud/google_cloud_pipeline_components/preview/automl/vision

1 file changed

+3
-1
lines changed

components/google-cloud/google_cloud_pipeline_components/preview/automl/vision/data_converter.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
from kfp import dsl
2323

2424

25+
# pylint: disable=singleton-comparison
2526
# pylint: disable=g-doc-args
2627
@dsl.container_component
2728
def data_converter(
@@ -31,6 +32,7 @@ def data_converter(
3132
objective: str,
3233
output_dir: dsl.Output[dsl.Artifact],
3334
gcp_resources: dsl.OutputPath(str),
35+
enable_input_validation: bool = True,
3436
location: str = 'us-central1',
3537
timeout: str = '604800s',
3638
service_account: Optional[str] = None,
@@ -75,7 +77,7 @@ def data_converter(
7577
'image_uri': 'us-docker.pkg.dev/vertex-ai/vertex-vision-model-garden-dockers/data-converter',
7678
'args': [
7779
'--enable_input_validation',
78-
'true',
80+
str(enable_input_validation),
7981
'--input_file_path',
8082
input_file_path,
8183
'--input_file_type',

0 commit comments

Comments
 (0)