Skip to content

Commit 2ea84ee

Browse files
committed
Moved dtype setting function to finalize output function
1 parent 9c3a2fb commit 2ea84ee

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pandas/io/parsers/arrow_parser_wrapper.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -217,6 +217,9 @@ def _finalize_pandas_output(self, frame: DataFrame) -> DataFrame:
217217
except TypeError as err:
218218
# GH#44901 reraise to keep api consistent
219219
raise ValueError(str(err)) from err
220+
221+
self._set_date_column_dtype(frame, date_columns=["date_column1", "date_column2"], dtype="timestamp[ns][pyarrow]")
222+
220223
return frame
221224

222225
def _validate_usecols(self, usecols) -> None:
@@ -306,8 +309,6 @@ def read(self) -> DataFrame:
306309

307310
else:
308311
frame = table.to_pandas()
309-
310-
self._set_date_column_dtype(frame, date_columns=["date_column1", "date_column2"], dtype="timestamp[ns][pyarrow]")
311312

312313
return self._finalize_pandas_output(frame)
313314

0 commit comments

Comments
 (0)