fix prediction format issue

googleapis · Jun 3, 2022 · a44d055 · a44d055
1 parent 6022bba
commit a44d055
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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=(