From 2505a8ae53bccba199e30333a3d42c6af2771272 Mon Sep 17 00:00:00 2001 From: Brock Date: Tue, 16 Mar 2021 09:59:23 -0700 Subject: [PATCH] pre-commit fixups --- pandas/core/arrays/timedeltas.py | 8 +++++--- pandas/tests/arrays/test_datetimelike.py | 4 ++-- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/pandas/core/arrays/timedeltas.py b/pandas/core/arrays/timedeltas.py index f86d8755acb22..f3889ff360aa8 100644 --- a/pandas/core/arrays/timedeltas.py +++ b/pandas/core/arrays/timedeltas.py @@ -42,7 +42,6 @@ from pandas.core.dtypes.common import ( DT64NS_DTYPE, TD64NS_DTYPE, - is_categorical_dtype, is_dtype_equal, is_float_dtype, is_integer_dtype, @@ -53,7 +52,10 @@ pandas_dtype, ) from pandas.core.dtypes.dtypes import DatetimeTZDtype -from pandas.core.dtypes.generic import ABCMultiIndex +from pandas.core.dtypes.generic import ( + ABCCategorical, + ABCMultiIndex, +) from pandas.core.dtypes.missing import isna from pandas.core import nanops @@ -970,7 +972,7 @@ def sequence_to_td64ns( elif not isinstance(data, (np.ndarray, ExtensionArray)): # GH#24539 e.g. xarray, dask object data = np.asarray(data) - elif is_categorical_dtype(data.dtype): + elif isinstance(data, ABCCategorical): data = data.categories.take(data.codes, fill_value=NaT)._values copy = False diff --git a/pandas/tests/arrays/test_datetimelike.py b/pandas/tests/arrays/test_datetimelike.py index 1ee366d452f5a..1a7e2d1d820f7 100644 --- a/pandas/tests/arrays/test_datetimelike.py +++ b/pandas/tests/arrays/test_datetimelike.py @@ -1372,9 +1372,9 @@ def array_likes(request): data = memoryview(arr) elif name == "array": # stdlib array - from array import array + from array import array as array_stdlib - data = array("i", arr) + data = array_stdlib("i", arr) elif name == "dask": import dask.array