annotate-magics

Author: twoertwein

pandas/_testing/contexts.py

@@ -5,6 +5,7 @@
 from pathlib import Path
 from shutil import rmtree
 import tempfile
+from types import TracebackType
 from typing import (
     IO,
     Any,
@@ -237,6 +238,11 @@ def __enter__(self) -> None:
         self.start_state = np.random.get_state()
         np.random.seed(self.seed)
 
-    def __exit__(self, exc_type, exc_value, traceback) -> None:
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> None:
 
         np.random.set_state(self.start_state)

pandas/conftest.py

@@ -515,7 +515,7 @@ def __getitem__(self, key):
         def __iter__(self):
             return self._data.__iter__()
 
-        def __len__(self):
+        def __len__(self) -> int:
             return self._data.__len__()
 
     return TestNonDictMapping

pandas/core/arrays/datetimelike.py

@@ -1937,7 +1937,9 @@ class TimelikeOps(DatetimeLikeArrayMixin):
 
     _default_dtype: np.dtype
 
-    def __init__(self, values, dtype=None, freq=lib.no_default, copy: bool = False):
+    def __init__(
+        self, values, dtype=None, freq=lib.no_default, copy: bool = False
+    ) -> None:
         values = extract_array(values, extract_numpy=True)
         if isinstance(values, IntegerArray):
             values = values.to_numpy("int64", na_value=iNaT)

pandas/core/dtypes/base.py

@@ -403,7 +403,7 @@ def __init__(self, storage=None) -> None:
     def __repr__(self) -> str:
         return f"{self.name}[{self.storage}]"
 
-    def __str__(self):
+    def __str__(self) -> str:
         return self.name
 
     def __eq__(self, other: Any) -> bool:

pandas/core/internals/concat.py

@@ -359,7 +359,7 @@ def _get_mgr_concatenation_plan(mgr: BlockManager, indexers: dict[int, np.ndarra
 
 
 class JoinUnit:
-    def __init__(self, block: Block, shape: Shape, indexers=None):
+    def __init__(self, block: Block, shape: Shape, indexers=None) -> None:
         # Passing shape explicitly is required for cases when block is None.
         # Note: block is None implies indexers is None, but not vice-versa
         if indexers is None:

pandas/core/strings/object_array.py

@@ -32,7 +32,7 @@ class ObjectStringArrayMixin(BaseStringArrayMethods):
 
     _str_na_value = np.nan
 
-    def __len__(self):
+    def __len__(self) -> int:
         # For typing, _str_map relies on the object being sized.
         raise NotImplementedError
 

pandas/io/excel/_base.py

@@ -7,6 +7,7 @@
 from io import BytesIO
 import os
 from textwrap import fill
+from types import TracebackType
 from typing import (
     IO,
     Any,
@@ -1449,7 +1450,12 @@ def check_extension(cls, ext: str) -> Literal[True]:
     def __enter__(self) -> ExcelWriter:
         return self
 
-    def __exit__(self, exc_type, exc_value, traceback) -> None:
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> None:
         self.close()
 
     def close(self) -> None:
@@ -1746,7 +1752,12 @@ def close(self) -> None:
     def __enter__(self) -> ExcelFile:
         return self
 
-    def __exit__(self, exc_type, exc_value, traceback) -> None:
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> None:
         self.close()
 
     def __del__(self) -> None:

pandas/io/json/_json.py

@@ -8,6 +8,7 @@
 import functools
 from io import StringIO
 from itertools import islice
+from types import TracebackType
 from typing import (
     TYPE_CHECKING,
     Any,
@@ -991,7 +992,12 @@ def __next__(self) -> DataFrame | Series:
     def __enter__(self) -> JsonReader[FrameSeriesStrT]:
         return self
 
-    def __exit__(self, exc_type, exc_value, traceback) -> None:
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> None:
         self.close()
 
 

pandas/io/parsers/readers.py

@@ -8,6 +8,7 @@
 import inspect
 import sys
 from textwrap import fill
+from types import TracebackType
 from typing import (
     IO,
     Any,
@@ -1806,7 +1807,12 @@ def get_chunk(self, size: int | None = None) -> DataFrame:
     def __enter__(self) -> TextFileReader:
         return self
 
-    def __exit__(self, exc_type, exc_value, traceback) -> None:
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> None:
         self.close()
 
 

pandas/io/pytables.py

@@ -15,6 +15,7 @@
 import os
 import re
 from textwrap import dedent
+from types import TracebackType
 from typing import (
     TYPE_CHECKING,
     Any,
@@ -632,7 +633,12 @@ def __repr__(self) -> str:
     def __enter__(self) -> HDFStore:
         return self
 
-    def __exit__(self, exc_type, exc_value, traceback) -> None:
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> None:
         self.close()
 
     def keys(self, include: str = "pandas") -> list[str]:

pandas/io/sas/sasreader.py

@@ -7,6 +7,7 @@
     ABCMeta,
     abstractmethod,
 )
+from types import TracebackType
 from typing import (
     TYPE_CHECKING,
     Hashable,
@@ -48,7 +49,12 @@ def close(self) -> None:
     def __enter__(self) -> ReaderBase:
         return self
 
-    def __exit__(self, exc_type, exc_value, traceback) -> None:
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> None:
         self.close()
 
 

pandas/io/stata.py

@@ -18,6 +18,7 @@
 import os
 import struct
 import sys
+from types import TracebackType
 from typing import (
     IO,
     TYPE_CHECKING,
@@ -1178,7 +1179,12 @@ def __enter__(self) -> StataReader:
         """enter context manager"""
         return self
 
-    def __exit__(self, exc_type, exc_value, traceback) -> None:
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> None:
         """exit context manager"""
         self.close()
 

pandas/tests/dtypes/test_inference.py

@@ -103,7 +103,7 @@ def it_outer():
 
         return it_outer()
 
-    def __len__(self):
+    def __len__(self) -> int:
         return len(self._values)
 
     def __array__(self, t=None):
@@ -1954,7 +1954,7 @@ class Numeric(Number):
             def __init__(self, value) -> None:
                 self.value = value
 
-            def __int__(self):
+            def __int__(self) -> int:
                 return self.value
 
         num = Numeric(1)

pandas/tests/extension/arrow/arrays.py

@@ -91,7 +91,7 @@ def _from_sequence(cls, values, dtype=None, copy=False):
         arr = pa.chunked_array([pa.array(np.asarray(values))])
         return cls(arr)
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         return f"{type(self).__name__}({repr(self._data)})"
 
     def __contains__(self, obj) -> bool:

pandas/tests/frame/test_arithmetic.py

@@ -2032,7 +2032,7 @@ def _constructor_expanddim(self):
     class SubclassedDataFrame(DataFrame):
         _metadata = ["my_extra_data"]
 
-        def __init__(self, my_extra_data, *args, **kwargs):
+        def __init__(self, my_extra_data, *args, **kwargs) -> None:
             self.my_extra_data = my_extra_data
             super().__init__(*args, **kwargs)
 

pandas/tests/frame/test_constructors.py

@@ -1394,7 +1394,7 @@ def __init__(self, lst) -> None:
             def __getitem__(self, n):
                 return self._lst.__getitem__(n)
 
-            def __len__(self):
+            def __len__(self) -> int:
                 return self._lst.__len__()
 
         lst_containers = [DummyContainer([1, "a"]), DummyContainer([2, "b"])]

pandas/tests/io/formats/test_format.py

@@ -2816,7 +2816,7 @@ def name(self):
                 return "DtypeStub"
 
         class ExtTypeStub(pd.api.extensions.ExtensionArray):
-            def __len__(self):
+            def __len__(self) -> int:
                 return 2
 
             def __getitem__(self, ix):