fix(components): GCP - AutoML - Fixed input type in Batch predict (#4499

)
kubeflow · Sep 16, 2020 · 41ce68e · 41ce68e
1 parent 3fd72e4
commit 41ce68e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/components/gcp/automl/prediction_service_batch_predict/component.py b/components/gcp/automl/prediction_service_batch_predict/component.py
@@ -17,7 +17,7 @@
 
 def automl_prediction_service_batch_predict(
     model_path,
-    gcs_input_uris: str = None,
+    gcs_input_uris: list = None,
     gcs_output_uri_prefix: str = None,
     bq_input_uri: str = None,
     bq_output_uri: str = None,

diff --git a/components/gcp/automl/prediction_service_batch_predict/component.yaml b/components/gcp/automl/prediction_service_batch_predict/component.yaml
@@ -2,7 +2,7 @@ name: Automl prediction service batch predict
 inputs:
 - name: model_path
 - name: gcs_input_uris
-  type: String
+  type: JsonArray
   optional: true
 - name: gcs_output_uri_prefix
   type: String
@@ -90,7 +90,7 @@ implementation:
       _missing_arg = object()
       _parser = argparse.ArgumentParser(prog='Automl prediction service batch predict', description='')
       _parser.add_argument("--model-path", dest="model_path", type=str, required=True, default=_missing_arg)
-      _parser.add_argument("--gcs-input-uris", dest="gcs_input_uris", type=str, required=False, default=_missing_arg)
+      _parser.add_argument("--gcs-input-uris", dest="gcs_input_uris", type=json.loads, required=False, default=_missing_arg)
       _parser.add_argument("--gcs-output-uri-prefix", dest="gcs_output_uri_prefix", type=str, required=False, default=_missing_arg)
       _parser.add_argument("--bq-input-uri", dest="bq_input_uri", type=str, required=False, default=_missing_arg)
       _parser.add_argument("--bq-output-uri", dest="bq_output_uri", type=str, required=False, default=_missing_arg)