Add exists_ok=True back

Signed-off-by: Tsotne Tabidze <tsotne@tecton.ai>
feast-dev · May 20, 2021 · 43f4c08 · 43f4c08
1 parent 2837dd9
commit 43f4c08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdk/python/feast/infra/offline_stores/bigquery.py b/sdk/python/feast/infra/offline_stores/bigquery.py
@@ -212,7 +212,7 @@ def _upload_entity_df_into_bigquery(project, dataset_name, entity_df, client) ->
         client.get_dataset(dataset)
     except NotFound:
         # Only create the dataset if it does not exist
-        client.create_dataset(dataset)
+        client.create_dataset(dataset, exists_ok=True)
 
     # Drop the index so that we dont have unnecessary columns
     entity_df.reset_index(drop=True, inplace=True)