Skip to content

Commit d3fbf3a

Browse files
authored
Chore: small cleanup of dialect.py (#909)
1 parent 48eaba3 commit d3fbf3a

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

sqlmesh/core/dialect.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ def _parse_lambda(self: Parser, alias: bool = False) -> t.Optional[exp.Expressio
8989

9090
def _parse_macro(self: Parser, keyword_macro: str = "") -> t.Optional[exp.Expression]:
9191
index = self._index
92-
field = self._parse_primary() or self._parse_function({}) or self._parse_id_var()
92+
field = self._parse_primary() or self._parse_function(functions={}) or self._parse_id_var()
9393

9494
if isinstance(field, exp.Func):
9595
macro_name = field.name.upper()
@@ -98,12 +98,11 @@ def _parse_macro(self: Parser, keyword_macro: str = "") -> t.Optional[exp.Expres
9898
return None
9999

100100
if isinstance(field, exp.Anonymous):
101-
name = field.name.upper()
102-
if name == "DEF":
101+
if macro_name == "DEF":
103102
return self.expression(
104103
MacroDef, this=field.expressions[0], expression=field.expressions[1]
105104
)
106-
if name == "SQL":
105+
if macro_name == "SQL":
107106
into = field.expressions[1].this.lower() if len(field.expressions) > 1 else None
108107
return self.expression(MacroSQL, this=field.expressions[0], into=into)
109108

@@ -256,6 +255,7 @@ def parse(self: Parser) -> t.Optional[exp.Expression]:
256255
props = self._parse_wrapped_csv(functools.partial(_parse_props, self))
257256
else:
258257
props = None
258+
259259
value = self.expression(
260260
ModelKind,
261261
this=kind.value,
@@ -526,7 +526,7 @@ def pandas_to_sql(
526526
columns_to_types: t.Optional[t.Dict[str, exp.DataType]] = None,
527527
batch_size: int = 0,
528528
alias: str = "t",
529-
) -> t.Generator[exp.Select, None, None]:
529+
) -> t.Iterator[exp.Select]:
530530
"""Convert a pandas dataframe into a VALUES sql statement.
531531
532532
Args:

0 commit comments

Comments
 (0)