File tree Expand file tree Collapse file tree 2 files changed +2
-3
lines changed
Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -925,8 +925,7 @@ def generate_test(
925925 cte_output = test ._execute (cte_query )
926926 ctes [cte .alias ] = (
927927 pandas_timestamp_to_pydatetime (
928- cte_output .apply (lambda col : col .map (_normalize_df_value )),
929- cte_query .named_selects ,
928+ df = cte_output .apply (lambda col : col .map (_normalize_df_value )),
930929 )
931930 .replace ({np .nan : None })
932931 .to_dict (orient = "records" )
Original file line number Diff line number Diff line change @@ -444,7 +444,7 @@ def to_time_column(
444444
445445
446446def pandas_timestamp_to_pydatetime (
447- df : pd .DataFrame , columns_to_types : t .Optional [t .Dict [str , exp .DataType ]]
447+ df : pd .DataFrame , columns_to_types : t .Optional [t .Dict [str , exp .DataType ]] = None
448448) -> pd .DataFrame :
449449 import pandas as pd
450450 from pandas .api .types import is_datetime64_any_dtype # type: ignore
You can’t perform that action at this time.
0 commit comments