From 786b12f57a9aa34ddf12b5f6ba1eeac65884697d Mon Sep 17 00:00:00 2001 From: Iaroslav Zeigerman Date: Sat, 7 Dec 2024 20:14:45 -0800 Subject: [PATCH] Address comments --- sqlmesh/utils/pandas.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sqlmesh/utils/pandas.py b/sqlmesh/utils/pandas.py index d19a19af8..bab58a21d 100644 --- a/sqlmesh/utils/pandas.py +++ b/sqlmesh/utils/pandas.py @@ -36,7 +36,7 @@ def columns_to_types_from_df(df: pd.DataFrame) -> t.Dict[str, exp.DataType]: for column_name, column_type in df.dtypes.items(): exp_type: t.Optional[exp.DataType] = None if hasattr(pd, "DatetimeTZDtype") and isinstance(column_type, pd.DatetimeTZDtype): - exp_type = exp.DataType.build("timestamp") + exp_type = exp.DataType.build("timestamptz") else: exp_type = PANDAS_TYPE_MAPPINGS.get(column_type) if not exp_type: