Skip to content

Commit b7f9bd1

Browse files
authored
fix: address issues with not updating tables properly (#6391)
* fix: address issues with not updating tables properly This is a hotfix it's probably not very performant? * fix: indents * fix: explicitly set if exists on drop
1 parent 476056a commit b7f9bd1

File tree

1 file changed

+13
-10
lines changed

1 file changed

+13
-10
lines changed

warehouse/scheduler/scheduler/mq/handlers/data_model.py

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -254,18 +254,21 @@ async def evaluate_single_model(
254254

255255
resolved_query = await model.resolve_query(table_resolvers=table_resolvers)
256256

257-
create_query = ctas_query(resolved_query)
257+
with adapter.transaction():
258+
# Delete existing table if it exists
259+
adapter.drop_table(table_name=target_table, exists=True)
258260

259-
adapter.ctas(
260-
table_name=target_table,
261-
query_or_df=create_query,
262-
exists=True,
263-
)
261+
create_query = ctas_query(resolved_query)
264262

265-
adapter.replace_query(
266-
table_name=target_table,
267-
query_or_df=resolved_query,
268-
)
263+
adapter.ctas(
264+
table_name=target_table,
265+
query_or_df=create_query,
266+
exists=True,
267+
)
268+
adapter.replace_query(
269+
table_name=target_table,
270+
query_or_df=resolved_query,
271+
)
269272

270273
columns = adapter.columns(table_name=target_table)
271274

0 commit comments

Comments
 (0)