Skip to content

Commit a99c08a

Browse files
committed
Change dict to set as only fqns needed
1 parent bccd583 commit a99c08a

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

sqlmesh/dbt/basemodel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -333,7 +333,7 @@ def sqlmesh_model_kwargs(
333333
{
334334
source.canonical_name(context)
335335
for source in model_context.sources.values()
336-
if source.fqn not in context.models_by_fqn
336+
if source.fqn not in context.model_fqns
337337
# Allow dbt projects to reference a model as a source without causing a cycle
338338
},
339339
),

sqlmesh/dbt/context.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ class DbtContext:
5151
_project_name: t.Optional[str] = None
5252
_variables: t.Dict[str, t.Any] = field(default_factory=dict)
5353
_models: t.Dict[str, ModelConfig] = field(default_factory=dict)
54-
_models_by_fqn: t.Dict[str, ModelConfig] = field(default_factory=dict)
54+
_model_fqns: t.Set[str] = field(default_factory=set)
5555
_seeds: t.Dict[str, SeedConfig] = field(default_factory=dict)
5656
_sources: t.Dict[str, SourceConfig] = field(default_factory=dict)
5757
_refs: t.Dict[str, t.Union[ModelConfig, SeedConfig]] = field(default_factory=dict)
@@ -145,7 +145,7 @@ def models(self) -> t.Dict[str, ModelConfig]:
145145
def models(self, models: t.Dict[str, ModelConfig]) -> None:
146146
self._models = {}
147147
self._refs = {}
148-
self._models_by_fqn = {}
148+
self._model_fqns = set()
149149
self.add_models(models)
150150

151151
def add_models(self, models: t.Dict[str, ModelConfig]) -> None:
@@ -154,10 +154,10 @@ def add_models(self, models: t.Dict[str, ModelConfig]) -> None:
154154
self._jinja_environment = None
155155

156156
@property
157-
def models_by_fqn(self) -> t.Dict[str, ModelConfig]:
158-
if not self._models_by_fqn:
159-
self._models_by_fqn = {model.fqn: model for model in self._models.values()}
160-
return self._models_by_fqn
157+
def model_fqns(self) -> t.Set[str]:
158+
if not self._model_fqns:
159+
self._model_fqns = {model.fqn for model in self._models.values()}
160+
return self._model_fqns
161161

162162
@property
163163
def seeds(self) -> t.Dict[str, SeedConfig]:

0 commit comments

Comments
 (0)