From 963dfb083543c22b90ca36ccd1902d885334b983 Mon Sep 17 00:00:00 2001 From: Bu Sun Kim Date: Thu, 13 Aug 2020 05:37:10 +0000 Subject: [PATCH] chore: fix batch predict test --- automl/beta/batch_predict.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/automl/beta/batch_predict.py b/automl/beta/batch_predict.py index 911eec733cbb..8dd5acefa283 100644 --- a/automl/beta/batch_predict.py +++ b/automl/beta/batch_predict.py @@ -38,9 +38,10 @@ def batch_predict( output_config = automl.types.BatchPredictOutputConfig( gcs_destination=gcs_destination ) + params = {} response = prediction_client.batch_predict( - model_full_id, input_config, output_config + model_full_id, input_config, output_config, params=params ) print("Waiting for operation to complete...")