diff --git a/samples/tables/automl_tables_predict.py b/samples/tables/automl_tables_predict.py index 61643ed6..069faac1 100644 --- a/samples/tables/automl_tables_predict.py +++ b/samples/tables/automl_tables_predict.py @@ -72,9 +72,9 @@ def batch_predict(project_id, client = automl.TablesClient(project=project_id, region=compute_region) # Query model - response = client.batch_predict( - gcs_input_uris, gcs_output_uri, model_display_name=model_display_name - ) + response = client.batch_predict(gcs_input_uris=gcs_input_uris, + gcs_output_uri_prefix=gcs_output_uri, + model_display_name=model_display_name) print("Making batch prediction... ") response.result() print("Batch prediction complete.\n{}".format(response.metadata))