ANN206 for pyi

Author: cmp0xff

pandas-stubs/core/arrays/interval.pyi

@@ -5,11 +5,9 @@ from typing import (
 )
 
 import numpy as np
-from pandas import (
-    Index,
-    Series,
-)
 from pandas.core.arrays.base import ExtensionArray as ExtensionArray
+from pandas.core.indexes.base import Index
+from pandas.core.series import Series
 from typing_extensions import Self
 
 from pandas._libs.interval import (
@@ -32,15 +30,15 @@ class IntervalArray(IntervalMixin, ExtensionArray):
     can_hold_na: bool = ...
     def __new__(
         cls, data, closed=..., dtype=..., copy: bool = ..., verify_integrity: bool = ...
-    ): ...
+    ) -> Self: ...
     @classmethod
     def from_breaks(
         cls,
         breaks,
         closed: str = "right",
         copy: bool = False,
         dtype=None,
-    ): ...
+    ) -> Self: ...
     @classmethod
     def from_arrays(
         cls,
@@ -49,15 +47,15 @@ class IntervalArray(IntervalMixin, ExtensionArray):
         closed: str = "right",
         copy: bool = False,
         dtype=...,
-    ): ...
+    ) -> Self: ...
     @classmethod
     def from_tuples(
         cls,
         data,
         closed: str = "right",
         copy: bool = False,
         dtype=None,
-    ): ...
+    ) -> Self: ...
     def __array__(
         self, dtype: NpDtype | None = None, copy: bool | None = None
     ) -> np_1darray: ...

pandas-stubs/core/arrays/sparse/accessor.pyi

@@ -1,5 +1,6 @@
-from pandas import Series
 from pandas.core.accessor import PandasDelegate
+from pandas.core.series import Series
+from typing_extensions import Self
 
 class BaseAccessor:
     def __init__(self, data=...) -> None: ...
@@ -12,7 +13,7 @@ class SparseAccessor(BaseAccessor, PandasDelegate):
 
 class SparseFrameAccessor(BaseAccessor, PandasDelegate):
     @classmethod
-    def from_spmatrix(cls, data, index=..., columns=...): ...
+    def from_spmatrix(cls, data, index=..., columns=...) -> Self: ...
     def to_dense(self): ...
     def to_coo(self): ...
     @property

pandas-stubs/core/arrays/sparse/array.pyi

@@ -33,7 +33,7 @@ class SparseArray(ExtensionArray, ExtensionOpsMixin):
         copy: bool = ...,
     ) -> None: ...
     @classmethod
-    def from_spmatrix(cls, data): ...
+    def from_spmatrix(cls, data) -> Self: ...
     def __array__(
         self, dtype: NpDtype | None = None, copy: bool | None = None
     ) -> np_1darray: ...

pandas-stubs/core/indexes/range.pyi

@@ -13,6 +13,7 @@ from pandas.core.indexes.base import (
     Index,
     _IndexSubclassBase,
 )
+from typing_extensions import Self
 
 from pandas._typing import (
     HashableT,
@@ -32,7 +33,7 @@ class RangeIndex(_IndexSubclassBase[int, np.int64]):
         name: Hashable = ...,
     ): ...
     @classmethod
-    def from_range(cls, data, name: Hashable = ..., dtype=...): ...
+    def from_range(cls, data, name: Hashable = ..., dtype=...) -> Self: ...
     def __reduce__(self): ...
     @property
     def start(self) -> int: ...

pyproject.toml

@@ -201,7 +201,7 @@ ignore = [
   "A004",    # https://docs.astral.sh/ruff/rules/builtin-import-shadowing/
   "PYI001",  # https://docs.astral.sh/ruff/rules/unprefixed-type-param/
   "PYI042",  # https://docs.astral.sh/ruff/rules/snake-case-type-alias/
-  "ERA001", "ANN001", "ANN201", "ANN204", "ANN206", "PLR0402", "PLC0105"
+  "ERA001", "ANN001", "ANN201", "ANN204", "PLR0402", "PLC0105"
   ]
 "scripts/*" = [
   # The following rules are ignored permanently for good reasons