Add IF EXISTs clause for partitioned and mat view tables (#4133)

So that dev deployments that aren't partitioned can run
bcgov · Nov 25, 2024 · 8185c92 · 8185c92
1 parent 3cfabd7
commit 8185c92
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/alembic/versions/4014ddf1f874_partition_weather_station_model_.py b/api/alembic/versions/4014ddf1f874_partition_weather_station_model_.py
@@ -17,8 +17,8 @@
 
 def upgrade():
     # ### drop table that's now partitioned ###
-    op.execute("DROP MATERIALIZED VIEW morecast_2_materialized_view")
-    op.drop_table("weather_station_model_predictions_retired")
+    op.execute("DROP MATERIALIZED VIEW IF EXISTS morecast_2_materialized_view")
+    op.drop_table("weather_station_model_predictions_retired", if_exists=True)
 
 
 def downgrade():