From 6a06b0750cdcb12387e7fab14de2c7de455aedb0 Mon Sep 17 00:00:00 2001
From: Michael Hu <michaelanhu@gmail.com>
Date: Fri, 3 Jun 2022 12:06:35 -0400
Subject: [PATCH] fix prediction format issue

---
 tests/system/aiplatform/test_e2e_forecasting.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/system/aiplatform/test_e2e_forecasting.py b/tests/system/aiplatform/test_e2e_forecasting.py
index 3f2dd6c9464..18c981b9fe5 100644
--- a/tests/system/aiplatform/test_e2e_forecasting.py
+++ b/tests/system/aiplatform/test_e2e_forecasting.py
@@ -108,6 +108,7 @@ def test_end_to_end_forecasting(self, shared_state, training_job):
                 job_display_name=self._make_display_name(
                     "forecasting-liquor-model"),
                 instances_format="bigquery",
+                predictions_format="csv",
                 machine_type="n1-standard-4",
                 bigquery_source=_PREDICTION_DATASET_BQ_PATH,
                 gcs_destination_prefix=(