@@ -89,7 +89,7 @@ def _parse_lambda(self: Parser, alias: bool = False) -> t.Optional[exp.Expressio
8989
9090def _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