File tree Expand file tree Collapse file tree 2 files changed +17
-1
lines changed
Expand file tree Collapse file tree 2 files changed +17
-1
lines changed Original file line number Diff line number Diff line change @@ -357,7 +357,9 @@ def sqlmesh_model_kwargs(
357357 # - https://docs.getdbt.com/reference/resource-configs/contract
358358 # - https://docs.getdbt.com/reference/resource-configs/column_types
359359 if column_types_override :
360- model_kwargs ["columns" ] = column_types_to_sqlmesh (column_types_override )
360+ model_kwargs ["columns" ] = (
361+ column_types_to_sqlmesh (column_types_override , self .dialect (context )) or None
362+ )
361363
362364 return model_kwargs
363365
Original file line number Diff line number Diff line change 77from dbt .adapters .base import BaseRelation , Column
88from pytest_mock import MockerFixture
99
10+ from sqlglot import exp
1011from sqlmesh .core .audit import StandaloneAudit
1112from sqlmesh .core .config import Config , ModelDefaultsConfig
1213from sqlmesh .core .dialect import jinja_query
1314from sqlmesh .core .model import SqlModel
1415from sqlmesh .core .model .kind import OnDestructiveChange , OnAdditiveChange
16+ from sqlmesh .dbt .column import ColumnConfig
1517from sqlmesh .dbt .common import Dependencies
1618from sqlmesh .dbt .context import DbtContext
1719from sqlmesh .dbt .loader import sqlmesh_config
@@ -1076,3 +1078,15 @@ def test_on_schema_change_properties(
10761078
10771079 assert model .on_additive_change == expected_additive
10781080 assert model .on_destructive_change == expected_destructive
1081+
1082+
1083+ def test_sqlmesh_model_kwargs_columns_override ():
1084+ context = DbtContext ()
1085+ context .project_name = "Foo"
1086+ context .target = DuckDbConfig (name = "target" , schema = "foo" )
1087+
1088+ kwargs = ModelConfig (dialect = "duckdb" ).sqlmesh_model_kwargs (
1089+ context ,
1090+ {"c" : ColumnConfig (name = "c" , data_type = "uinteger" )},
1091+ )
1092+ assert kwargs .get ("columns" ) == {"c" : exp .DataType .build (exp .DataType .Type .UINT )}
You can’t perform that action at this time.
0 commit comments