Skip to content

Commit

Permalink
feat(components): internal
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 614085882
  • Loading branch information
Googler committed Apr 24, 2024
1 parent ae530de commit ef4cd70
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@
from kfp import dsl


# pylint: disable=singleton-comparison
# pylint: disable=g-doc-args
@dsl.container_component
def data_converter(
Expand All @@ -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,
Expand Down Expand Up @@ -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',
str(enable_input_validation),
'--input_file_path',
input_file_path,
'--input_file_type',
Expand Down

0 comments on commit ef4cd70

Please # to comment.