annotate-imprecise-magics

Author: twoertwein

pandas/conftest.py

@@ -29,7 +29,10 @@
 from decimal import Decimal
 import operator
 import os
-from typing import Callable
+from typing import (
+    Callable,
+    Iterator,
+)
 
 from dateutil.tz import (
     tzlocal,
@@ -512,7 +515,7 @@ def __init__(self, underlying_dict) -> None:
         def __getitem__(self, key):
             return self._data.__getitem__(key)
 
-        def __iter__(self):
+        def __iter__(self) -> Iterator:
             return self._data.__iter__()
 
         def __len__(self) -> int:

pandas/core/arrays/categorical.py

@@ -8,6 +8,7 @@
 from typing import (
     TYPE_CHECKING,
     Hashable,
+    Iterator,
     Literal,
     Sequence,
     TypeVar,
@@ -2093,7 +2094,7 @@ def take_nd(
         )
         return self.take(indexer, allow_fill=allow_fill, fill_value=fill_value)
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         """
         Returns an Iterator over the values of this Categorical.
         """

pandas/core/arrays/datetimelike.py

@@ -10,6 +10,7 @@
     TYPE_CHECKING,
     Any,
     Callable,
+    Iterator,
     Literal,
     Sequence,
     TypeVar,
@@ -287,7 +288,7 @@ def _box_values(self, values) -> np.ndarray:
         """
         return lib.map_infer(values, self._box_func, convert=False)
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         if self.ndim > 1:
             return (self[n] for n in range(len(self)))
         else:

pandas/core/arrays/datetimes.py

@@ -9,6 +9,7 @@
 import inspect
 from typing import (
     TYPE_CHECKING,
+    Iterator,
     Literal,
     cast,
 )
@@ -567,7 +568,7 @@ def __array__(self, dtype=None) -> np.ndarray:
 
         return super().__array__(dtype=dtype)
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         """
         Return an iterator over the boxed values
 

pandas/core/arrays/interval.py

@@ -8,6 +8,7 @@
 import textwrap
 from typing import (
     TYPE_CHECKING,
+    Iterator,
     Literal,
     Sequence,
     TypeVar,
@@ -652,7 +653,7 @@ def size(self) -> int:
     # ---------------------------------------------------------------------
     # EA Interface
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         return iter(np.asarray(self))
 
     def __len__(self) -> int:

pandas/core/arrays/masked.py

@@ -3,6 +3,7 @@
 from typing import (
     TYPE_CHECKING,
     Any,
+    Iterator,
     Literal,
     Sequence,
     TypeVar,
@@ -239,7 +240,7 @@ def __setitem__(self, key, value) -> None:
         self._data[key] = value
         self._mask[key] = mask
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         if self.ndim == 1:
             for i in range(len(self)):
                 if self._mask[i]:

pandas/core/arrays/timedeltas.py

@@ -3,6 +3,7 @@
 from datetime import timedelta
 from typing import (
     TYPE_CHECKING,
+    Iterator,
     cast,
 )
 
@@ -304,7 +305,7 @@ def astype(self, dtype, copy: bool = True):
 
         return dtl.DatetimeLikeArrayMixin.astype(self, dtype, copy=copy)
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         if self.ndim > 1:
             for i in range(len(self)):
                 yield self[i]

pandas/core/base.py

@@ -11,6 +11,7 @@
     Any,
     Generic,
     Hashable,
+    Iterator,
     Literal,
     TypeVar,
     cast,
@@ -739,7 +740,7 @@ def tolist(self):
 
     to_list = tolist
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         """
         Return an iterator of the values.
 

pandas/core/computation/ops.py

@@ -10,6 +10,7 @@
 from typing import (
     Callable,
     Iterable,
+    Iterator,
     Literal,
 )
 
@@ -214,7 +215,7 @@ def __init__(self, op: str, operands: Iterable[Term | Op], encoding=None) -> Non
         self.operands = operands
         self.encoding = encoding
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         return iter(self.operands)
 
     def __repr__(self) -> str:

pandas/core/generic.py

@@ -16,6 +16,7 @@
     Callable,
     ClassVar,
     Hashable,
+    Iterator,
     Literal,
     Mapping,
     NoReturn,
@@ -1951,7 +1952,7 @@ def _drop_labels_or_levels(self, keys, axis: int = 0):
     # "object" defined the type as "Callable[[object], int]")
     __hash__: ClassVar[None]  # type: ignore[assignment]
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         """
         Iterate over info axis.
 

pandas/core/groupby/grouper.py

@@ -9,6 +9,7 @@
     TYPE_CHECKING,
     Any,
     Hashable,
+    Iterator,
     final,
 )
 import warnings
@@ -564,7 +565,7 @@ def __init__(
     def __repr__(self) -> str:
         return f"Grouping({self.name})"
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         return iter(self.indices)
 
     @cache_readonly

pandas/core/groupby/ops.py

@@ -1310,7 +1310,7 @@ def _sort_idx(self) -> npt.NDArray[np.intp]:
         # Counting sort indexer
         return get_group_index_sorter(self.labels, self.ngroups)
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         sdata = self.sorted_data
 
         if self.ngroups == 0:

pandas/core/strings/accessor.py

@@ -8,6 +8,7 @@
     TYPE_CHECKING,
     Callable,
     Hashable,
+    Iterator,
     cast,
 )
 import warnings
@@ -239,7 +240,7 @@ def __getitem__(self, key):
         result = self._data.array._str_getitem(key)
         return self._wrap_result(result)
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         warnings.warn(
             "Columnar iteration over characters will be deprecated in future releases.",
             FutureWarning,

pandas/core/window/rolling.py

@@ -14,6 +14,7 @@
     Any,
     Callable,
     Hashable,
+    Iterator,
     Sized,
 )
 import warnings
@@ -369,7 +370,7 @@ def __repr__(self) -> str:
         attrs = ",".join(attrs_list)
         return f"{type(self).__name__} [{attrs}]"
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         obj = self._selected_obj.set_axis(self._on)
         obj = self._create_data(obj)
         indexer = self._get_window_indexer()

pandas/io/pytables.py

@@ -1909,7 +1909,7 @@ def __init__(
 
         self.auto_close = auto_close
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         # iterate
         current = self.start
         if self.coordinates is None:
@@ -2131,7 +2131,7 @@ def cvalues(self):
         """return my cython values"""
         return self.values
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         return iter(self.values)
 
     def maybe_set_size(self, min_itemsize=None) -> None:

pandas/tests/dtypes/test_inference.py

@@ -18,6 +18,7 @@
 from numbers import Number
 import re
 import sys
+from typing import Iterator
 
 import numpy as np
 import pytest
@@ -95,7 +96,7 @@ class MockNumpyLikeArray:
     def __init__(self, values) -> None:
         self._values = values
 
-    def __iter__(self):
+    def __iter__(self) -> Iterator:
         iter_values = iter(self._values)
 
         def it_outer():

pandas/tests/frame/constructors/test_from_records.py

@@ -1,5 +1,6 @@
 from datetime import datetime
 from decimal import Decimal
+from typing import Iterator
 
 import numpy as np
 import pytest
@@ -200,7 +201,7 @@ def __init__(self, *args) -> None:
             def __getitem__(self, i):
                 return self.args[i]
 
-            def __iter__(self):
+            def __iter__(self) -> Iterator:
                 return iter(self.args)
 
         recs = [Record(1, 2, 3), Record(4, 5, 6), Record(7, 8, 9)]

pandas/tests/frame/test_constructors.py

@@ -10,6 +10,7 @@
 import functools
 import itertools
 import re
+from typing import Iterator
 import warnings
 
 import numpy as np
@@ -1430,7 +1431,7 @@ def test_constructor_list_of_ranges(self):
     def test_constructor_iterable(self):
         # GH 21987
         class Iter:
-            def __iter__(self):
+            def __iter__(self) -> Iterator:
                 for i in range(10):
                     yield [1, 2, 3]
 

pandas/tests/io/json/test_readlines.py

@@ -1,5 +1,6 @@
 from io import StringIO
 from pathlib import Path
+from typing import Iterator
 
 import pytest
 
@@ -289,7 +290,7 @@ def read(self, *args):
             self.read_count += 1
             return self.stringio.read(*args)
 
-        def __iter__(self):
+        def __iter__(self) -> Iterator:
             self.read_count += 1
             return iter(self.stringio)
 

pandas/tests/io/parser/test_python_parser_only.py

@@ -11,6 +11,7 @@
     BytesIO,
     StringIO,
 )
+from typing import Iterator
 
 import pytest
 
@@ -322,7 +323,7 @@ class NoNextBuffer:
         def __init__(self, csv_data) -> None:
             self.data = csv_data
 
-        def __iter__(self):
+        def __iter__(self) -> Iterator:
             return self.data.__iter__()
 
         def read(self):

pandas/tests/io/test_html.py

@@ -8,6 +8,7 @@
 from pathlib import Path
 import re
 import threading
+from typing import Iterator
 from urllib.error import URLError
 
 import numpy as np
@@ -1277,7 +1278,7 @@ def seek(self, offset):
             def seekable(self):
                 return True
 
-            def __iter__(self):
+            def __iter__(self) -> Iterator:
                 # to fool `is_file_like`, should never end up here
                 assert False
 

pandas/tests/reshape/concat/test_concat.py

@@ -3,6 +3,7 @@
     deque,
 )
 from decimal import Decimal
+from typing import Iterator
 from warnings import (
     catch_warnings,
     simplefilter,
@@ -463,7 +464,7 @@ def __getitem__(self, index):
         tm.assert_frame_equal(concat(CustomIterator1(), ignore_index=True), expected)
 
         class CustomIterator2(abc.Iterable):
-            def __iter__(self):
+            def __iter__(self) -> Iterator:
                 yield df1
                 yield df2
 

pandas/tests/series/test_constructors.py

@@ -3,6 +3,7 @@
     datetime,
     timedelta,
 )
+from typing import Iterator
 
 from dateutil.tz import tzoffset
 import numpy as np
@@ -252,7 +253,7 @@ def test_constructor_series(self):
     def test_constructor_iterable(self):
         # GH 21987
         class Iter:
-            def __iter__(self):
+            def __iter__(self) -> Iterator:
                 yield from range(10)
 
         expected = Series(list(range(10)), dtype="int64")