fix unions

tobymao · tobymao · commit a8c259d5e1ba · 2023-12-18T20:09:50.000-08:00
diff --git a/sqlmesh/core/model/definition.py b/sqlmesh/core/model/definition.py
@@ -1008,7 +1008,7 @@ def columns_to_types(self) -> t.Optional[t.Dict[str, exp.DataType]]:
 
             self._columns_to_types = {
                 select.output_name: select.type or exp.DataType.build("unknown")
-                for select in query.expressions
+                for select in query.selects
             }
 
         if "*" in self._columns_to_types:

Original file line number	Diff line number	Diff line change
`@@ -1008,7 +1008,7 @@ def columns_to_types(self) -> t.Optional[t.Dict[str, exp.DataType]]:`
`1008`	`1008`
`1009`	`1009`	`self._columns_to_types = {`
`1010`	`1010`	`select.output_name: select.type or exp.DataType.build("unknown")`
`1011`		`- for select in query.expressions`
	`1011`	`+ for select in query.selects`
`1012`	`1012`	`}`
`1013`	`1013`
`1014`	`1014`	`if "*" in self._columns_to_types:`