🎨 sparse: IndexMixin code cleanup

Author: jorenham

scipy-stubs/sparse/_index.pyi

@@ -11,13 +11,16 @@ from ._typing import Complex, Float, Int, Scalar, Shape
 _SCT = TypeVar("_SCT", bound=Scalar, default=Any)
 _ShapeT_co = TypeVar("_ShapeT_co", bound=Shape, default=Shape, covariant=True)
 
-_Self1DT = TypeVar("_Self1DT", bound=IndexMixin[Any, tuple[int]])
-_Self2DT = TypeVar("_Self2DT", bound=IndexMixin[Any, tuple[int, int]])
+_Self1DT = TypeVar("_Self1DT", bound=IndexMixin[Any, _1D])
+_Self2DT = TypeVar("_Self2DT", bound=IndexMixin[Any, _2D])
 
 _ToInt: TypeAlias = str | op.CanInt | op.CanIndex | op.CanTrunc | Buffer
 _ToFloat: TypeAlias = str | op.CanFloat | op.CanIndex | Buffer
 _ToComplex: TypeAlias = str | op.CanComplex | op.CanFloat | op.CanIndex | complex
 
+_1D: TypeAlias = tuple[int]  # noqa: PYI042
+_2D: TypeAlias = tuple[int, int]  # noqa: PYI042
+
 _ToIndex1D: TypeAlias = op.CanIndex | tuple[op.CanIndex]
 _ToIndex2D: TypeAlias = tuple[op.CanIndex, op.CanIndex]
 
@@ -31,7 +34,7 @@ _ToSlice2D: TypeAlias = (
     | list[bool]
     | list[np.bool_]
     | list[int]
-    | _spbase[np.bool_, tuple[int, int]]
+    | _spbase[np.bool_, _2D]
 )
 
 ###
@@ -40,30 +43,30 @@ INT_TYPES: tuple[type[int], type[np.integer[Any]]] = ...
 
 class IndexMixin(Generic[_SCT, _ShapeT_co]):
     @overload
-    def __getitem__(self: IndexMixin[Any, tuple[int]], ix: op.CanIndex, /) -> _SCT: ...
+    def __getitem__(self: IndexMixin[Any, _1D], ix: op.CanIndex, /) -> _SCT: ...
     @overload
-    def __getitem__(self: IndexMixin[Any, tuple[int, int]], ix: _ToIndex2D, /) -> _SCT: ...
+    def __getitem__(self: IndexMixin[Any, _2D], ix: _ToIndex2D, /) -> _SCT: ...
     @overload
     def __getitem__(self: _Self1DT, ixs: _ToSlice1D, /) -> _Self1DT: ...
     @overload
     def __getitem__(self: _Self2DT, ixs: _ToSlice2D, /) -> _Self2DT: ...
 
     #
     @overload
-    def __setitem__(self: IndexMixin[Any, tuple[int]], key: op.CanIndex | _ToSlice1D, x: _SCT, /) -> None: ...
+    def __setitem__(self: IndexMixin[Any, _1D], ix: op.CanIndex | _ToSlice1D, x: _SCT, /) -> None: ...
     @overload
-    def __setitem__(self: IndexMixin[Any], key: _ToIndex2D | _ToSlice2D, x: _SCT, /) -> None: ...
+    def __setitem__(self: IndexMixin[Any], ix: _ToIndex2D | _ToSlice2D, x: _SCT, /) -> None: ...
     @overload
-    def __setitem__(self: IndexMixin[Int, tuple[int]], key: _ToIndex1D, x: _ToInt, /) -> None: ...
+    def __setitem__(self: IndexMixin[Int, _1D], ix: _ToIndex1D, x: _ToInt, /) -> None: ...
     @overload
-    def __setitem__(self: IndexMixin[Int], key: _ToIndex2D, x: _ToInt, /) -> None: ...
+    def __setitem__(self: IndexMixin[Int], ix: _ToIndex2D, x: _ToInt, /) -> None: ...
     @overload
-    def __setitem__(self: IndexMixin[Float, tuple[int]], key: _ToIndex1D, x: _ToFloat, /) -> None: ...
+    def __setitem__(self: IndexMixin[Float, _1D], ix: _ToIndex1D, x: _ToFloat, /) -> None: ...
     @overload
-    def __setitem__(self: IndexMixin[Float], key: _ToIndex2D, x: _ToFloat, /) -> None: ...
+    def __setitem__(self: IndexMixin[Float], ix: _ToIndex2D, x: _ToFloat, /) -> None: ...
     @overload
-    def __setitem__(self: IndexMixin[Complex, tuple[int]], key: _ToIndex1D, x: _ToComplex, /) -> None: ...
+    def __setitem__(self: IndexMixin[Complex, _1D], ix: _ToIndex1D, x: _ToComplex, /) -> None: ...
     @overload
-    def __setitem__(self: IndexMixin[Complex], key: _ToIndex2D, x: _ToComplex, /) -> None: ...
+    def __setitem__(self: IndexMixin[Complex], ix: _ToIndex2D, x: _ToComplex, /) -> None: ...
     @overload
-    def __setitem__(self, key: _ToIndex1D | _ToSlice1D | _ToIndex2D | _ToSlice2D, x: object, /) -> None: ...
+    def __setitem__(self, ix: _ToIndex1D | _ToSlice1D | _ToIndex2D | _ToSlice2D, x: object, /) -> None: ...