Skip to content

Commit 6cf1de3

Browse files
pr feedback
1 parent 257ef63 commit 6cf1de3

File tree

4 files changed

+10
-22
lines changed

4 files changed

+10
-22
lines changed

sqlmesh/core/model/kind.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -916,12 +916,6 @@ def data_hash_values(self) -> t.List[t.Optional[str]]:
916916
self.dialect,
917917
]
918918

919-
@property
920-
def metadata_hash_values(self) -> t.List[t.Optional[str]]:
921-
return [
922-
*super().metadata_hash_values,
923-
]
924-
925919
def to_expression(
926920
self, expressions: t.Optional[t.List[exp.Expression]] = None, **kwargs: t.Any
927921
) -> d.ModelKind:

sqlmesh/core/snapshot/evaluator.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@
8383
format_additive_change_msg,
8484
AdditiveChangeError,
8585
)
86-
from sqlmesh.utils.jinja import JinjaMacroRegistry, MacroReturnVal
86+
from sqlmesh.utils.jinja import MacroReturnVal
8787

8888
if sys.version_info >= (3, 12):
8989
from importlib import metadata
@@ -2685,9 +2685,7 @@ def _execute_materialization(
26852685
create_only: bool = False,
26862686
**kwargs: t.Any,
26872687
) -> None:
2688-
from sqlmesh.dbt.builtin import create_builtin_globals
2689-
2690-
jinja_macros = getattr(model, "jinja_macros", JinjaMacroRegistry())
2688+
jinja_macros = model.jinja_macros
26912689
existing_globals = jinja_macros.global_objs.copy()
26922690

26932691
# For vdes we need to use the table, since we don't know the schema/table at parse time
@@ -2709,8 +2707,8 @@ def _execute_materialization(
27092707
"execution_dt": kwargs.get("execution_time"),
27102708
}
27112709

2712-
context = create_builtin_globals(
2713-
jinja_macros=jinja_macros, jinja_globals=jinja_globals, engine_adapter=self.adapter
2710+
context = jinja_macros._create_builtin_globals(
2711+
{"engine_adapter": self.adapter, **jinja_globals}
27142712
)
27152713

27162714
context.update(
@@ -2731,7 +2729,7 @@ def _execute_materialization(
27312729
try:
27322730
template.render(**context)
27332731
except MacroReturnVal as ret:
2734-
# this is a succesful return from a macro call (dbt uses this list of Relations to update their relation cache)
2732+
# this is a successful return from a macro call (dbt uses this list of Relations to update their relation cache)
27352733
returned_relations = ret.value.get("relations", [])
27362734
logger.info(
27372735
f"Materialization {self.materialization_name} returned relations: {returned_relations}"

sqlmesh/dbt/manifest.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ def __init__(
136136

137137
self._on_run_start_per_package: t.Dict[str, HookConfigs] = defaultdict(dict)
138138
self._on_run_end_per_package: t.Dict[str, HookConfigs] = defaultdict(dict)
139-
self._materializations_per_package: t.Dict[str, MaterializationConfigs] = defaultdict(dict)
139+
self._materializations: MaterializationConfigs = {}
140140

141141
def tests(self, package_name: t.Optional[str] = None) -> TestConfigs:
142142
self._load_all()
@@ -166,9 +166,9 @@ def on_run_end(self, package_name: t.Optional[str] = None) -> HookConfigs:
166166
self._load_all()
167167
return self._on_run_end_per_package[package_name or self._project_name]
168168

169-
def materializations(self, package_name: t.Optional[str] = None) -> MaterializationConfigs:
169+
def materializations(self) -> MaterializationConfigs:
170170
self._load_all()
171-
return self._materializations_per_package[package_name or self._project_name]
171+
return self._materializations
172172

173173
@property
174174
def all_macros(self) -> t.Dict[str, t.Dict[str, MacroInfo]]:
@@ -315,9 +315,7 @@ def _load_materializations(self) -> None:
315315
)
316316

317317
key = f"{mat_name}_{adapter}"
318-
self._materializations_per_package[macro.package_name][key] = (
319-
materialization_config
320-
)
318+
self._materializations[key] = materialization_config
321319

322320
def _load_tests(self) -> None:
323321
for node in self._manifest.nodes.values():

sqlmesh/dbt/package.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -105,9 +105,7 @@ def load(self, package_root: Path) -> Package:
105105
models = _fix_paths(self._context.manifest.models(package_name), package_root)
106106
seeds = _fix_paths(self._context.manifest.seeds(package_name), package_root)
107107
macros = _fix_paths(self._context.manifest.macros(package_name), package_root)
108-
materializations = _fix_paths(
109-
self._context.manifest.materializations(package_name), package_root
110-
)
108+
materializations = _fix_paths(self._context.manifest.materializations(), package_root)
111109
on_run_start = _fix_paths(self._context.manifest.on_run_start(package_name), package_root)
112110
on_run_end = _fix_paths(self._context.manifest.on_run_end(package_name), package_root)
113111
sources = self._context.manifest.sources(package_name)

0 commit comments

Comments
 (0)