Skip to content

Commit

Permalink
Fixed the type of the max_worker_count param for MLEngine Batch Pred (#…
Browse files Browse the repository at this point in the history
…12)

The type for the `max_worker_count` parameter for the ML Engine Batch Prediction
operator was set to `integer`, matching the airflow doc on the operator, but the
Google API expects a string value (a string formatted int64 to be precise).

The changes here fix the type mismatch issue and add typing to the input_paths
parameter's items, which was previously missing.
  • Loading branch information
kzvezdarov authored and mchalek committed Nov 27, 2018
1 parent 4a9e133 commit 01e8a5d
Showing 1 changed file with 3 additions and 3 deletions.
Expand Up @@ -33,8 +33,8 @@ parameters_jsonschema:
- TF_RECORD_GZIP
input_paths:
type: array
items:
type: string
items:
type: string
output_path:
type: string
region:
Expand All @@ -46,7 +46,7 @@ parameters_jsonschema:
uri:
type: string
max_worker_count:
type: integer
type: string
runtime_version:
type: string
gcp_conn_id:
Expand Down

0 comments on commit 01e8a5d

Please sign in to comment.