fixes #1958

francoisserra · francoisserra · commit a26d0e04c7a0 · 2021-11-04T11:05:04.000+01:00
Signed-off-by: François Serra &lt;francois.serra@adeo.com&gt;
diff --git a/sdk/python/feast/infra/offline_stores/bigquery.py b/sdk/python/feast/infra/offline_stores/bigquery.py
@@ -133,7 +133,7 @@ def get_historical_features(
         assert isinstance(config.offline_store, BigQueryOfflineStoreConfig)
 
         table_reference = _get_table_reference_for_new_entity(
-            client, client.project, config.offline_store.dataset
+            client, client.project, config.offline_store.dataset, config.offline_store.location
         )
 
         @contextlib.contextmanager
@@ -332,13 +332,13 @@ def _wait_until_done(bq_job):
 
 
 def _get_table_reference_for_new_entity(
-    client: Client, dataset_project: str, dataset_name: str
+    client: Client, dataset_project: str, dataset_name: str, dataset_location: str
 ) -> str:
     """Gets the table_id for the new entity to be uploaded."""
 
     # First create the BigQuery dataset if it doesn't exist
     dataset = bigquery.Dataset(f"{dataset_project}.{dataset_name}")
-    dataset.location = "US"
+    dataset.location = dataset_location
 
     try:
         client.get_dataset(dataset)
