From 4a9bcb622471ccd62bb042d05ff339b9898c5c26 Mon Sep 17 00:00:00 2001 From: Natalia Mokeeva Date: Mon, 27 Mar 2023 23:45:30 +0200 Subject: [PATCH 1/3] TYP: remove ignore from array_manager.py --- pandas/core/internals/array_manager.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/pandas/core/internals/array_manager.py b/pandas/core/internals/array_manager.py index 0408dfd83fedc..1365858c5316f 100644 --- a/pandas/core/internals/array_manager.py +++ b/pandas/core/internals/array_manager.py @@ -842,9 +842,7 @@ def iset( else: assert isinstance(loc, np.ndarray) assert loc.dtype == "bool" - # error: Incompatible types in assignment (expression has type "ndarray", - # variable has type "range") - indices = np.nonzero(loc)[0] # type: ignore[assignment] + indices = np.array([index for index in loc[1:]]) assert value.ndim == 2 assert value.shape[0] == len(self._axes[0]) From 31287f0e473526e8ba4f6e0f821e0cab65110671 Mon Sep 17 00:00:00 2001 From: Natalia Mokeeva Date: Tue, 28 Mar 2023 16:06:37 +0200 Subject: [PATCH 2/3] TYP: remove ignore from array_manager.py II --- pandas/core/internals/array_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pandas/core/internals/array_manager.py b/pandas/core/internals/array_manager.py index 513bc5c28eea0..c480c93e92535 100644 --- a/pandas/core/internals/array_manager.py +++ b/pandas/core/internals/array_manager.py @@ -842,7 +842,7 @@ def iset( else: assert isinstance(loc, np.ndarray) assert loc.dtype == "bool" - indices = np.array([index for index in loc[1:]]) + indices = np.array([i for i in range(len(loc)) if loc[i] is True]) assert value.ndim == 2 assert value.shape[0] == len(self._axes[0]) From c861a746c7a226480c32c8f96b59f5b9f909fead Mon Sep 17 00:00:00 2001 From: Natalia Mokeeva Date: Wed, 29 Mar 2023 11:44:34 +0200 Subject: [PATCH 3/3] TYP: remove ignore from array_manager.py III --- pandas/core/internals/array_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pandas/core/internals/array_manager.py b/pandas/core/internals/array_manager.py index c480c93e92535..0cc3e0d1ee5ae 100644 --- a/pandas/core/internals/array_manager.py +++ b/pandas/core/internals/array_manager.py @@ -834,7 +834,7 @@ def iset( # multiple columns -> convert slice or array to integer indices elif isinstance(loc, slice): - indices = range( + indices: range | np.ndarray = range( loc.start if loc.start is not None else 0, loc.stop if loc.stop is not None else self.shape_proper[1], loc.step if loc.step is not None else 1,