update bq uris

TheMichaelHu · TheMichaelHu · commit 604e85a8c107 · 2022-05-27T11:30:37.000-04:00
diff --git a/tests/system/aiplatform/test_e2e_forecasting.py b/tests/system/aiplatform/test_e2e_forecasting.py
@@ -22,10 +22,10 @@
 from tests.system.aiplatform import e2e_base
 
 _TRAINING_DATASET_BQ_PATH = (
-    "bq://bigquery-public-data:iowa_liquor_sales_forecasting.2020_sales_train"
+    "bq://ucaip-sample-tests:ucaip_test_us_central1.2020_sales_train"
 )
 _PREDICTION_DATASET_BQ_PATH = (
-    "bq://bigquery-public-data:iowa_liquor_sales_forecasting.2021_sales_predict"
+    "bq://ucaip-sample-tests:ucaip_test_us_central1.2021_sales_predict"
 )
 
 

Original file line number	Diff line number	Diff line change
`@@ -22,10 +22,10 @@`
`22`	`22`	`from tests.system.aiplatform import e2e_base`
`23`	`23`
`24`	`24`	`_TRAINING_DATASET_BQ_PATH = (`
`25`		`- "bq://bigquery-public-data:iowa_liquor_sales_forecasting.2020_sales_train"`
	`25`	`+ "bq://ucaip-sample-tests:ucaip_test_us_central1.2020_sales_train"`
`26`	`26`	`)`
`27`	`27`	`_PREDICTION_DATASET_BQ_PATH = (`
`28`		`- "bq://bigquery-public-data:iowa_liquor_sales_forecasting.2021_sales_predict"`
	`28`	`+ "bq://ucaip-sample-tests:ucaip_test_us_central1.2021_sales_predict"`
`29`	`29`	`)`
`30`	`30`
`31`	`31`