GoogleCloudPlatform / vertex-pipelines-end-to-end-samples

Apache License 2.0
225 stars 89 forks source link

model_batch_predict component doesn't support key_field for bigquery? #46

Open sprydog2964 opened 1 year ago

sprydog2964 commented 1 year ago

I'm trying to use the model_batch_predict component for prediction using bigquery table as input source. The component works in the pipeline. However, when I set

instancesConfig={'keyField': 'key'} # 'key' is an added column for referencing. 

it throws an error says: image

Does anyone have an idea how I can make it work?

felix-datatonic commented 10 months ago

Hi @sprydog2964 ,

Thanks for this issue. Please check out the most recent PR and release.

Do you mind trying the new release and let us know if the issue persists? We've successfully tested different instanceConfig as inputs to the components as documented here.