From a27bc1fc929f562faee128698c3d33a369e39b82 Mon Sep 17 00:00:00 2001 From: Luke Manley Date: Wed, 31 May 2023 21:45:08 -0400 Subject: [PATCH] TYP: algos.isin --- pandas/core/algorithms.py | 7 ++++--- pandas/core/tools/datetimes.py | 4 +--- pandas/io/parsers/base_parser.py | 6 +----- 3 files changed, 6 insertions(+), 11 deletions(-) diff --git a/pandas/core/algorithms.py b/pandas/core/algorithms.py index 33bd56e5b1068..9c2859c909b1b 100644 --- a/pandas/core/algorithms.py +++ b/pandas/core/algorithms.py @@ -83,6 +83,7 @@ if TYPE_CHECKING: from pandas._typing import ( + ListLike, NumpySorter, NumpyValueArrayLike, ) @@ -438,14 +439,14 @@ def unique_with_mask(values, mask: npt.NDArray[np.bool_] | None = None): unique1d = unique -def isin(comps: AnyArrayLike, values: AnyArrayLike) -> npt.NDArray[np.bool_]: +def isin(comps: ListLike, values: ListLike) -> npt.NDArray[np.bool_]: """ Compute the isin boolean array. Parameters ---------- - comps : array-like - values : array-like + comps : list-like + values : list-like Returns ------- diff --git a/pandas/core/tools/datetimes.py b/pandas/core/tools/datetimes.py index 74210a1ce5ad8..cba47c1f759b1 100644 --- a/pandas/core/tools/datetimes.py +++ b/pandas/core/tools/datetimes.py @@ -1252,9 +1252,7 @@ def calc_with_mask(carg, mask): # string with NaN-like try: - # error: Argument 2 to "isin" has incompatible type "List[Any]"; expected - # "Union[Union[ExtensionArray, ndarray], Index, Series]" - mask = ~algorithms.isin(arg, list(nat_strings)) # type: ignore[arg-type] + mask = ~algorithms.isin(arg, list(nat_strings)) return calc_with_mask(arg, mask) except (ValueError, OverflowError, TypeError): pass diff --git a/pandas/io/parsers/base_parser.py b/pandas/io/parsers/base_parser.py index f79b7ebe7efef..6cded2e21a43c 100644 --- a/pandas/io/parsers/base_parser.py +++ b/pandas/io/parsers/base_parser.py @@ -558,11 +558,7 @@ def _convert_to_ndarrays( try: values = lib.map_infer(values, conv_f) except ValueError: - # error: Argument 2 to "isin" has incompatible type "List[Any]"; - # expected "Union[Union[ExtensionArray, ndarray], Index, Series]" - mask = algorithms.isin( - values, list(na_values) # type: ignore[arg-type] - ).view(np.uint8) + mask = algorithms.isin(values, list(na_values)).view(np.uint8) values = lib.map_infer_mask(values, conv_f, mask) cvals, na_count = self._infer_types(