chore: fix build, ensure all dfs have unified date time types

etonlels · May 8, 2024 · c720b14 · c720b14
1 parent 914fa59
commit c720b14
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/core/engine_adapter/test_integration.py b/tests/core/engine_adapter/test_integration.py
@@ -110,11 +110,11 @@ def _init_engine_adapter(self) -> None:
 
     def _format_df(self, data: pd.DataFrame, to_datetime: bool = True) -> pd.DataFrame:
         for timestamp_column in self.timestamp_columns:
-            if timestamp_column in data.columns and to_datetime:
-                value = pd.to_datetime(data[timestamp_column])
-                if self.dialect == "duckdb":
-                    value = value.astype("datetime64[us]")
-                data[timestamp_column] = value
+            if timestamp_column in data.columns:
+                value = data[timestamp_column]
+                if to_datetime:
+                    value = pd.to_datetime(value)
+                data[timestamp_column] = value.astype("datetime64[ns]")
         return data
 
     def init(self):