From 917b6596dac1dfefbdd130944b393b1c2570f00f Mon Sep 17 00:00:00 2001 From: lafirm Date: Wed, 28 May 2025 11:55:18 +0530 Subject: [PATCH 1/2] add models only if not in projects --- sqlmesh/core/context.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/sqlmesh/core/context.py b/sqlmesh/core/context.py index 1f4722264c..7956e0baff 100644 --- a/sqlmesh/core/context.py +++ b/sqlmesh/core/context.py @@ -628,7 +628,9 @@ def load(self, update_schemas: bool = True) -> GenericContext[C]: for snapshot in self.state_reader.get_snapshots(prod.snapshots).values(): if snapshot.node.project in self._projects: uncached.add(snapshot.name) - else: + elif snapshot.name not in ( + set(project.models) | set(project.standalone_audits) + ): store = self._standalone_audits if snapshot.is_audit else self._models store[snapshot.name] = snapshot.node # type: ignore From 698d8f78fa266275046130132711ada75878fbf0 Mon Sep 17 00:00:00 2001 From: lafirm <136463254+lafirm@users.noreply.github.com> Date: Wed, 25 Jun 2025 11:32:33 +0530 Subject: [PATCH 2/2] restructure to add it to uncached (if exists both locally and in prod) --- sqlmesh/core/context.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/sqlmesh/core/context.py b/sqlmesh/core/context.py index 529e3dec6c..495c8c40f9 100644 --- a/sqlmesh/core/context.py +++ b/sqlmesh/core/context.py @@ -619,11 +619,12 @@ def load(self, update_schemas: bool = True) -> GenericContext[C]: for snapshot in self.state_reader.get_snapshots(prod.snapshots).values(): if snapshot.node.project in self._projects: uncached.add(snapshot.name) - elif snapshot.name not in ( - set(project.models) | set(project.standalone_audits) - ): - store = self._standalone_audits if snapshot.is_audit else self._models - store[snapshot.name] = snapshot.node # type: ignore + else: + local_store = self._standalone_audits if snapshot.is_audit else self._models + if snapshot.name in local_store: + uncached.add(snapshot.name) + else: + local_store[snapshot.name] = snapshot.node # type: ignore for model in self._models.values(): self.dag.add(model.fqn, model.depends_on)