Skip to content

Commit b578b6f

Browse files
authored
added duration if statement
1 parent eb5f8cf commit b578b6f

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

pandas/core/dtypes/dtypes.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2277,6 +2277,12 @@ def name(self) -> str: # type: ignore[override]
22772277
@cache_readonly
22782278
def numpy_dtype(self) -> np.dtype:
22792279
"""Return an instance of the related numpy dtype"""
2280+
if pa.types.is_duration(self.pyarrow_dtype):
2281+
# pa.duration(unit).to_pandas_dtype() returns ns units
2282+
# regardless of the pyarrow duration units
2283+
# This can be removed if/when pyarrow addresses it:
2284+
# https://github.com/apache/arrow/issues/34462
2285+
return np.dtype(f"timedelta64[{self.pyarrow_dtype.unit}]")
22802286
if pa.types.is_string(self.pyarrow_dtype) or pa.types.is_large_string(
22812287
self.pyarrow_dtype
22822288
):

0 commit comments

Comments
 (0)