Change drop to drop if exists

Signed-off-by: Felix Wang <wangfelix98@gmail.com>
feast-dev · Sep 30, 2021 · 4fa80bb · 4fa80bb
1 parent 93d5dd1
commit 4fa80bb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sdk/python/feast/infra/offline_stores/bigquery.py b/sdk/python/feast/infra/offline_stores/bigquery.py
@@ -160,7 +160,7 @@ def query_generator() -> Iterator[str]:
             finally:
                 # Asynchronously clean up the uploaded Bigquery table, which will expire
                 # if cleanup fails
-                client.query(query=f"DROP TABLE {table_reference}")
+                client.delete_table(table=table_reference, not_found_ok=True)
 
         return BigQueryRetrievalJob(
             query=query_generator,

diff --git a/sdk/python/feast/infra/offline_stores/redshift.py b/sdk/python/feast/infra/offline_stores/redshift.py
@@ -162,7 +162,7 @@ def query_generator() -> Iterator[str]:
                     config.offline_store.cluster_id,
                     config.offline_store.database,
                     config.offline_store.user,
-                    f"DROP TABLE {table_name}",
+                    f"DROP TABLE IF EXISTS {table_name}",
                 )
 
         return RedshiftRetrievalJob(