From 0953269c2633da301a4438f76aaeffc6c85410ee Mon Sep 17 00:00:00 2001 From: Trey Spiller Date: Wed, 22 Oct 2025 13:36:15 -0500 Subject: [PATCH] Fix pandas_timestamp_to_pydatetime call and arg defaults --- sqlmesh/core/test/definition.py | 3 +-- sqlmesh/utils/date.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/sqlmesh/core/test/definition.py b/sqlmesh/core/test/definition.py index 8d8ca17702..a3179ef6ee 100644 --- a/sqlmesh/core/test/definition.py +++ b/sqlmesh/core/test/definition.py @@ -925,8 +925,7 @@ def generate_test( cte_output = test._execute(cte_query) ctes[cte.alias] = ( pandas_timestamp_to_pydatetime( - cte_output.apply(lambda col: col.map(_normalize_df_value)), - cte_query.named_selects, + df=cte_output.apply(lambda col: col.map(_normalize_df_value)), ) .replace({np.nan: None}) .to_dict(orient="records") diff --git a/sqlmesh/utils/date.py b/sqlmesh/utils/date.py index 931cebf535..c9bb19c835 100644 --- a/sqlmesh/utils/date.py +++ b/sqlmesh/utils/date.py @@ -444,7 +444,7 @@ def to_time_column( def pandas_timestamp_to_pydatetime( - df: pd.DataFrame, columns_to_types: t.Optional[t.Dict[str, exp.DataType]] + df: pd.DataFrame, columns_to_types: t.Optional[t.Dict[str, exp.DataType]] = None ) -> pd.DataFrame: import pandas as pd from pandas.api.types import is_datetime64_any_dtype # type: ignore