File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed
components/google-cloud/google_cloud_pipeline_components/preview/automl/vision Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change 22
22
from kfp import dsl
23
23
24
24
25
+ # pylint: disable=singleton-comparison
25
26
# pylint: disable=g-doc-args
26
27
@dsl .container_component
27
28
def data_converter (
@@ -31,6 +32,7 @@ def data_converter(
31
32
objective : str ,
32
33
output_dir : dsl .Output [dsl .Artifact ],
33
34
gcp_resources : dsl .OutputPath (str ),
35
+ enable_input_validation : bool = True ,
34
36
location : str = 'us-central1' ,
35
37
timeout : str = '604800s' ,
36
38
service_account : Optional [str ] = None ,
@@ -75,7 +77,7 @@ def data_converter(
75
77
'image_uri' : 'us-docker.pkg.dev/vertex-ai/vertex-vision-model-garden-dockers/data-converter' ,
76
78
'args' : [
77
79
'--enable_input_validation' ,
78
- 'true' ,
80
+ str ( enable_input_validation ) ,
79
81
'--input_file_path' ,
80
82
input_file_path ,
81
83
'--input_file_type' ,
You can’t perform that action at this time.
0 commit comments