Skip to content

Commit 1261454

Browse files
authored
Fix: use dialect in column_to_types_ validator (#913)
* Fix: use dialect in column_to_types_ validator * Use DataType.build instead of maybe_parse
1 parent 245c76d commit 1261454

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

sqlmesh/core/model/meta.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
from croniter import croniter
88
from pydantic import Field, root_validator, validator
9-
from sqlglot import exp, maybe_parse
9+
from sqlglot import exp
1010

1111
from sqlmesh.core import dialect as d
1212
from sqlmesh.core.model.kind import (
@@ -194,12 +194,14 @@ def _cron_validator(cls, v: t.Any) -> t.Optional[str]:
194194
return cron
195195

196196
@validator("columns_to_types_", pre=True)
197-
def _columns_validator(cls, v: t.Any) -> t.Optional[t.Dict[str, exp.DataType]]:
197+
def _columns_validator(
198+
cls, v: t.Any, values: t.Dict[str, t.Any]
199+
) -> t.Optional[t.Dict[str, exp.DataType]]:
198200
if isinstance(v, exp.Schema):
199201
return {column.name: column.args["kind"] for column in v.expressions}
200202
if isinstance(v, dict):
201203
return {
202-
k: maybe_parse(data_type, into=exp.DataType) # type: ignore
204+
k: exp.DataType.build(data_type, dialect=values["dialect"])
203205
for k, data_type in v.items()
204206
}
205207
return v

0 commit comments

Comments
 (0)