From 8ac997cbb9db29bb49a8a117cc186015ed0a93c4 Mon Sep 17 00:00:00 2001 From: Willem Pienaar <6728866+woop@users.noreply.github.com> Date: Fri, 4 Jun 2021 14:17:59 -0700 Subject: [PATCH] Set BigQuery dataset for e2e tests (#1620) 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":