Skip to content

Commit 5720684

Browse files
committed
update dtype check
1 parent 75d1f32 commit 5720684

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

pandas/core/frame.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,6 @@
9393
is_array_like,
9494
is_bool_dtype,
9595
is_dataclass,
96-
is_datetime64_any_dtype,
9796
is_dict_like,
9897
is_float,
9998
is_float_dtype,
@@ -104,7 +103,6 @@
104103
is_list_like,
105104
is_scalar,
106105
is_sequence,
107-
is_timedelta64_dtype,
108106
needs_i8_conversion,
109107
pandas_dtype,
110108
)
@@ -11356,7 +11354,7 @@ def cov(
1135611354
"""
1135711355
data = self._get_numeric_data() if numeric_only else self
1135811356
dtypes = [blk.dtype for blk in self._mgr.blocks]
11359-
if any(is_datetime64_any_dtype(d) or is_timedelta64_dtype(d) for d in dtypes):
11357+
if any(d.kind in "mM" for d in dtypes):
1136011358
msg = (
1136111359
"DataFrame contains columns with dtype datetime64 "
1136211360
"or timedelta64, which are not supported for cov."

pandas/core/internals/managers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1800,7 +1800,7 @@ def as_array(
18001800
arr = np.asarray(blk.values, dtype=dtype)
18011801
else:
18021802
arr = np.array(blk.values, dtype=dtype, copy=copy)
1803-
if passed_nan and blk.dtype.kind in ["m", "M"]:
1803+
if passed_nan and blk.dtype.kind in "mM":
18041804
arr[isna(blk.values)] = na_value
18051805

18061806
if not copy:

0 commit comments

Comments
 (0)