From 23f74222b68b3b3c2fb9fa6ae9eca2eaeda950aa Mon Sep 17 00:00:00 2001 From: Willem Pienaar Date: Fri, 4 Jun 2021 13:35:11 -0700 Subject: [PATCH] Set BigQuery dataset for e2e tests Signed-off-by: Willem Pienaar --- sdk/python/tests/test_historical_retrieval.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/sdk/python/tests/test_historical_retrieval.py b/sdk/python/tests/test_historical_retrieval.py index 73f2dccaac..dc73a6832c 100644 --- a/sdk/python/tests/test_historical_retrieval.py +++ b/sdk/python/tests/test_historical_retrieval.py @@ -386,7 +386,9 @@ def test_historical_features_from_bigquery_sources( online_store=SqliteOnlineStoreConfig( path=os.path.join(temp_dir, "online_store.db"), ), - offline_store=BigQueryOfflineStoreConfig(type="bigquery",), + offline_store=BigQueryOfflineStoreConfig( + type="bigquery", dataset=bigquery_dataset + ), ) ) elif provider_type == "gcp": @@ -397,7 +399,9 @@ def test_historical_features_from_bigquery_sources( random.choices(string.ascii_uppercase + string.digits, k=10) ), provider="gcp", - offline_store=BigQueryOfflineStoreConfig(type="bigquery",), + offline_store=BigQueryOfflineStoreConfig( + type="bigquery", dataset=bigquery_dataset + ), ) ) elif provider_type == "gcp_custom_offline_config":