chore: fix batch predict test

googleapis · Aug 13, 2020 · f49335a · f49335a
1 parent 28697ed
commit f49335a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/samples/beta/batch_predict.py b/samples/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...")