Skip to content

Commit 78520de

Browse files
committed
Fix: Call on_complete handler when promoting / demoting external models
1 parent 4f68b36 commit 78520de

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

sqlmesh/core/snapshot/evaluator.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -425,7 +425,10 @@ def _promote_snapshot(
425425
on_complete: t.Optional[t.Callable[[SnapshotInfoLike], None]],
426426
) -> None:
427427
if snapshot.is_external:
428+
if on_complete is not None:
429+
on_complete(snapshot)
428430
return
431+
429432
qualified_view_name = snapshot.qualified_view_name
430433
schema = qualified_view_name.schema_for_environment(environment=environment)
431434
if schema is not None:
@@ -449,11 +452,10 @@ def _demote_snapshot(
449452
environment: str,
450453
on_complete: t.Optional[t.Callable[[SnapshotInfoLike], None]],
451454
) -> None:
452-
if snapshot.is_external:
453-
return
454-
view_name = snapshot.qualified_view_name.for_environment(environment=environment)
455-
logger.info("Dropping view '%s'", view_name)
456-
self.adapter.drop_view(view_name)
455+
if not snapshot.is_external:
456+
view_name = snapshot.qualified_view_name.for_environment(environment=environment)
457+
logger.info("Dropping view '%s'", view_name)
458+
self.adapter.drop_view(view_name)
457459

458460
if on_complete is not None:
459461
on_complete(snapshot)

0 commit comments

Comments
 (0)