feat(components): internal

PiperOrigin-RevId: 604508859
kubeflow · Apr 18, 2024 · f3f843f · f3f843f
1 parent d919ae7
commit f3f843f
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/...nts/google-cloud/google_cloud_pipeline_components/preview/automl/vision/data_converter.py b/...nts/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,
@@ -74,8 +76,6 @@ def data_converter(
               'container_spec': {
                   'image_uri': 'us-docker.pkg.dev/vertex-ai/vertex-vision-model-garden-dockers/data-converter',
                   'args': [
-                      '--enable_input_validation',
-                      'true',
                       '--input_file_path',
                       input_file_path,
                       '--input_file_type',
@@ -108,6 +108,12 @@ def data_converter(
       },
       'encryption_spec': {'kms_key_name': encryption_spec_key_name},
   }
+
+  if enable_input_validation == True:
+    payload['job_spec']['worker_pool_specs']['container_spec']['args'].append(
+        ['--enable_input_validation', 'true']
+    )
+
   return dsl.ContainerSpec(
       image=_image.GCPC_IMAGE_TAG,
       command=[