diff --git a/scipy-stubs/sparse/_bsr.pyi b/scipy-stubs/sparse/_bsr.pyi index f285707c..0cd44502 100644 --- a/scipy-stubs/sparse/_bsr.pyi +++ b/scipy-stubs/sparse/_bsr.pyi @@ -98,19 +98,84 @@ class bsr_array(_bsr_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d shape-like, dtype: None + @overload # 2-d shape-like, dtype: float64-like | None def __init__( self: bsr_array[np.float64], /, arg1: ToShape2D, - shape: None = None, - dtype: None = None, + shape: ToShape2D | None = None, + dtype: onp.AnyFloat64DType | None = None, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: bool-like (positional) + def __init__( + self: bsr_array[np.bool_], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyBoolDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: bool-like (keyword) + def __init__( + self: bsr_array[np.bool_], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyBoolDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: int-like (positional) + def __init__( + self: bsr_array[np.int64], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyIntDType, copy: bool = False, - blocksize: tuple[int, int] | None = None, *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.bool, dtype: type[bool] | None + @overload # 2-d shape-like, dtype: int-like (keyword) + def __init__( + self: bsr_array[np.int64], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyIntDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: complex128-like (positional) + def __init__( + self: bsr_array[np.complex128], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyComplex128DType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: complex128-like (keyword) + def __init__( + self: bsr_array[np.complex128], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyComplex128DType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # matrix-like builtins.bool, dtype: bool-like | None def __init__( self: bsr_array[np.bool_], /, @@ -122,7 +187,7 @@ class bsr_array(_bsr_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.int, dtype: type[int] | None + @overload # matrix-like builtins.int, dtype: int-like | None def __init__( self: bsr_array[np.int_], /, @@ -134,7 +199,7 @@ class bsr_array(_bsr_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.float, dtype: type[float] | None + @overload # matrix-like builtins.float, dtype: float64-like | None def __init__( self: bsr_array[np.float64], /, @@ -146,7 +211,7 @@ class bsr_array(_bsr_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.complex, dtype: type[complex] | None + @overload # matrix-like builtins.complex, dtype: complex128-like | None def __init__( self: bsr_array[np.complex128], /, @@ -236,7 +301,7 @@ class bsr_matrix(_bsr_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.bool, dtype: type[bool] | None + @overload # matrix-like builtins.bool, dtype: bool-like | None def __init__( self: bsr_matrix[np.bool_], /, @@ -248,7 +313,7 @@ class bsr_matrix(_bsr_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.int, dtype: type[int] | None + @overload # matrix-like builtins.int, dtype: int-like | None def __init__( self: bsr_matrix[np.int_], /, @@ -260,7 +325,7 @@ class bsr_matrix(_bsr_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.float, dtype: type[float] | None + @overload # matrix-like builtins.float, dtype: float64-like | None def __init__( self: bsr_matrix[np.float64], /, @@ -272,7 +337,7 @@ class bsr_matrix(_bsr_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.complex, dtype: type[complex] | None + @overload # matrix-like builtins.complex, dtype: complex128-like | None def __init__( self: bsr_matrix[np.complex128], /, diff --git a/scipy-stubs/sparse/_coo.pyi b/scipy-stubs/sparse/_coo.pyi index 83db8a44..52ebe0d5 100644 --- a/scipy-stubs/sparse/_coo.pyi +++ b/scipy-stubs/sparse/_coo.pyi @@ -149,7 +149,7 @@ class coo_array(_coo_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 1-d shape-like, dtype: None + @overload # 1-d shape-like, dtype: float64-like | None def __init__( self: coo_array[np.float64, tuple[int]], /, @@ -160,7 +160,7 @@ class coo_array(_coo_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d shape-like, dtype: None + @overload # 2-d shape-like, dtype: float64-like | None def __init__( self: coo_array[np.float64, tuple[int, int]], /, @@ -171,7 +171,7 @@ class coo_array(_coo_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # >2-d shape-like, dtype: None + @overload # >2-d shape-like, dtype: float64-like | None def __init__( self: coo_array[np.float64, onp.AtLeast3D], /, @@ -182,7 +182,205 @@ class coo_array(_coo_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # vector-like builtins.bool, dtype: type[bool] | None + @overload # 1-d shape-like, dtype: bool-like (positional) + def __init__( + self: coo_array[np.bool_, tuple[int]], + /, + arg1: ToShape1D, + shape: ToShape1D | None, + dtype: onp.AnyBoolDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: bool-like (keyword) + def __init__( + self: coo_array[np.bool_, tuple[int]], + /, + arg1: ToShape1D, + shape: ToShape1D | None = None, + *, + dtype: onp.AnyBoolDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: bool-like (positional) + def __init__( + self: coo_array[np.bool_, tuple[int, int]], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyBoolDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: bool-like (keyword) + def __init__( + self: coo_array[np.bool_, tuple[int, int]], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyBoolDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # >2-d shape-like, dtype: bool-like (positional) + def __init__( + self: coo_array[np.bool_, onp.AtLeast3D], + /, + arg1: ToShapeMin3D, + shape: ToShapeMin3D | None, + dtype: onp.AnyBoolDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # >2-d shape-like, dtype: bool-like (keyword) + def __init__( + self: coo_array[np.bool_, onp.AtLeast3D], + /, + arg1: ToShapeMin3D, + shape: ToShapeMin3D | None = None, + *, + dtype: onp.AnyBoolDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: int-like (positional) + def __init__( + self: coo_array[np.int64, tuple[int]], + /, + arg1: ToShape1D, + shape: ToShape1D | None, + dtype: onp.AnyIntDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: int-like (keyword) + def __init__( + self: coo_array[np.int64, tuple[int]], + /, + arg1: ToShape1D, + shape: ToShape1D | None = None, + *, + dtype: onp.AnyIntDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: int-like (positional) + def __init__( + self: coo_array[np.int64, tuple[int, int]], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyIntDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: int-like (keyword) + def __init__( + self: coo_array[np.int64, tuple[int, int]], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyIntDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # >2-d shape-like, dtype: int-like (positional) + def __init__( + self: coo_array[np.int64, onp.AtLeast3D], + /, + arg1: ToShapeMin3D, + shape: ToShapeMin3D | None, + dtype: onp.AnyIntDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # >2-d shape-like, dtype: int-like (keyword) + def __init__( + self: coo_array[np.int64, onp.AtLeast3D], + /, + arg1: ToShapeMin3D, + shape: ToShapeMin3D | None = None, + *, + dtype: onp.AnyIntDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: complex128-like (positional) + def __init__( + self: coo_array[np.complex128, tuple[int]], + /, + arg1: ToShape1D, + shape: ToShape1D | None, + dtype: onp.AnyComplex128DType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: complex128-like (keyword) + def __init__( + self: coo_array[np.complex128, tuple[int]], + /, + arg1: ToShape1D, + shape: ToShape1D | None = None, + *, + dtype: onp.AnyComplex128DType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: complex128-like (positional) + def __init__( + self: coo_array[np.complex128, tuple[int, int]], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyComplex128DType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: complex128-like (keyword) + def __init__( + self: coo_array[np.complex128, tuple[int, int]], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyComplex128DType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # >2-d shape-like, dtype: complex128-like (positional) + def __init__( + self: coo_array[np.complex128, onp.AtLeast3D], + /, + arg1: ToShapeMin3D, + shape: ToShapeMin3D | None, + dtype: onp.AnyComplex64DType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # >2-d shape-like, dtype: complex128-like (keyword) + def __init__( + self: coo_array[np.complex128, onp.AtLeast3D], + /, + arg1: ToShapeMin3D, + shape: ToShapeMin3D | None = None, + *, + dtype: onp.AnyComplex64DType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # vector-like builtins.bool, dtype: bool-like | None def __init__( self: coo_array[np.bool_, tuple[int]], /, @@ -193,7 +391,7 @@ class coo_array(_coo_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.bool, dtype: type[bool] | None + @overload # matrix-like builtins.bool, dtype: bool-like | None def __init__( self: coo_array[np.bool_, tuple[int, int]], /, @@ -204,7 +402,7 @@ class coo_array(_coo_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # vector-like builtins.int, dtype: type[int] | None + @overload # vector-like builtins.int, dtype: int-like | None def __init__( self: coo_array[np.int_, tuple[int]], /, @@ -215,7 +413,7 @@ class coo_array(_coo_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.int, dtype: type[int] | None + @overload # matrix-like builtins.int, dtype: int-like | None def __init__( self: coo_array[np.int_, tuple[int, int]], /, @@ -226,7 +424,7 @@ class coo_array(_coo_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # vector-like builtins.float, dtype: type[float] | None + @overload # vector-like builtins.float, dtype: float64-like | None def __init__( self: coo_array[np.float64, tuple[int]], /, @@ -237,7 +435,7 @@ class coo_array(_coo_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.float, dtype: type[float] | None + @overload # matrix-like builtins.float, dtype: float64-like | None def __init__( self: coo_array[np.float64, tuple[int, int]], /, @@ -248,7 +446,7 @@ class coo_array(_coo_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.complex, dtype: type[complex] | None + @overload # matrix-like builtins.complex, dtype: complex128-like | None def __init__( self: coo_array[np.complex128, tuple[int]], /, @@ -259,7 +457,7 @@ class coo_array(_coo_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.complex, dtype: type[complex] | None + @overload # matrix-like builtins.complex, dtype: complex128-like | None def __init__( self: coo_array[np.complex128, tuple[int, int]], /, @@ -403,7 +601,7 @@ class coo_matrix(_coo_base[_ScalarT_co, tuple[int, int]], spmatrix[_ScalarT_co], *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.bool, dtype: type[bool] | None + @overload # matrix-like builtins.bool, dtype: bool-like | None def __init__( self: coo_matrix[np.bool_], /, @@ -414,7 +612,7 @@ class coo_matrix(_coo_base[_ScalarT_co, tuple[int, int]], spmatrix[_ScalarT_co], *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.int, dtype: type[int] | None + @overload # matrix-like builtins.int, dtype: int-like | None def __init__( self: coo_matrix[np.int_], /, @@ -425,7 +623,7 @@ class coo_matrix(_coo_base[_ScalarT_co, tuple[int, int]], spmatrix[_ScalarT_co], *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.float, dtype: type[float] | None + @overload # matrix-like builtins.float, dtype: float64-like | None def __init__( self: coo_matrix[np.float64], /, @@ -436,7 +634,7 @@ class coo_matrix(_coo_base[_ScalarT_co, tuple[int, int]], spmatrix[_ScalarT_co], *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.complex, dtype: type[complex] | None + @overload # matrix-like builtins.complex, dtype: complex128-like | None def __init__( self: coo_matrix[np.complex128], /, diff --git a/scipy-stubs/sparse/_csc.pyi b/scipy-stubs/sparse/_csc.pyi index 9b3122ff..9ec43565 100644 --- a/scipy-stubs/sparse/_csc.pyi +++ b/scipy-stubs/sparse/_csc.pyi @@ -77,7 +77,7 @@ class csc_array(_csc_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d shape-like, dtype: None + @overload # 2-d shape-like, dtype: float64-like | None def __init__( self: csc_array[np.float64], /, @@ -88,7 +88,73 @@ class csc_array(_csc_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like bool, dtype: type[bool] | None + @overload # 2-d shape-like, dtype: bool-like (positional) + def __init__( + self: csc_array[np.bool_], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyBoolDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: bool-like (keyword) + def __init__( + self: csc_array[np.bool_], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyBoolDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: int-like (positional) + def __init__( + self: csc_array[np.int64], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyIntDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: int-like (keyword) + def __init__( + self: csc_array[np.int64], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyIntDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: complex128-like (positional) + def __init__( + self: csc_array[np.complex128], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyComplex128DType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: complex128-like (keyword) + def __init__( + self: csc_array[np.complex128], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyComplex128DType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d array-like bool, dtype: bool-like | None def __init__( self: csc_array[np.bool_], /, @@ -99,7 +165,7 @@ class csc_array(_csc_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~int, dtype: type[int] | None + @overload # 2-d array-like ~int, dtype: int-like | None def __init__( self: csc_array[np.int_], /, @@ -110,7 +176,7 @@ class csc_array(_csc_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~float, dtype: type[float] | None + @overload # 2-d array-like ~float, dtype: float64-like | None def __init__( self: csc_array[np.float64], /, @@ -121,7 +187,7 @@ class csc_array(_csc_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~complex, dtype: type[complex] | None + @overload # 2-d array-like ~complex, dtype: complex128-like | None def __init__( self: csc_array[np.complex128], /, @@ -200,7 +266,7 @@ class csc_matrix(_csc_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like bool, dtype: type[bool] | None + @overload # 2-d array-like bool, dtype: bool-like | None def __init__( self: csc_matrix[np.bool_], /, @@ -211,7 +277,7 @@ class csc_matrix(_csc_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~int, dtype: type[int] | None + @overload # 2-d array-like ~int, dtype: int-like | None def __init__( self: csc_matrix[np.int_], /, @@ -222,7 +288,7 @@ class csc_matrix(_csc_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~float, dtype: type[float] | None + @overload # 2-d array-like ~float, dtype: float64-like | None def __init__( self: csc_matrix[np.float64], /, @@ -233,7 +299,7 @@ class csc_matrix(_csc_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~complex, dtype: type[complex] | None + @overload # 2-d array-like ~complex, dtype: complex128-like | None def __init__( self: csc_matrix[np.complex128], /, diff --git a/scipy-stubs/sparse/_csr.pyi b/scipy-stubs/sparse/_csr.pyi index 0a24dd0c..5444255d 100644 --- a/scipy-stubs/sparse/_csr.pyi +++ b/scipy-stubs/sparse/_csr.pyi @@ -120,7 +120,7 @@ class csr_array(_csr_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 1-d shape-like, dtype: None + @overload # 1-d shape-like, dtype: float64-like | None def __init__( self: csr_array[np.float64, tuple[int]], /, @@ -131,7 +131,7 @@ class csr_array(_csr_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d shape-like, dtype: None + @overload # 2-d shape-like, dtype: float64-like | None def __init__( self: csr_array[np.float64, tuple[int, int]], /, @@ -142,7 +142,139 @@ class csr_array(_csr_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 1-d array-like bool, dtype: type[bool] | None + @overload # 1-d shape-like, dtype: bool-like (positional) + def __init__( + self: csr_array[np.bool_, tuple[int]], + /, + arg1: ToShape1D, + shape: ToShape1D | None, + dtype: onp.AnyBoolDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: bool-like (keyword) + def __init__( + self: csr_array[np.bool_, tuple[int]], + /, + arg1: ToShape1D, + shape: ToShape1D | None = None, + *, + dtype: onp.AnyBoolDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: bool-like (positional) + def __init__( + self: csr_array[np.bool_, tuple[int, int]], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyBoolDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: bool-like (keyword) + def __init__( + self: csr_array[np.bool_, tuple[int, int]], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyBoolDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: int-like (positional) + def __init__( + self: csr_array[np.int64, tuple[int]], + /, + arg1: ToShape1D, + shape: ToShape1D | None, + dtype: onp.AnyIntDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: int-like (keyword) + def __init__( + self: csr_array[np.int64, tuple[int]], + /, + arg1: ToShape1D, + shape: ToShape1D | None = None, + *, + dtype: onp.AnyIntDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: int-like (positional) + def __init__( + self: csr_array[np.int64, tuple[int, int]], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyIntDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: int-like (keyword) + def __init__( + self: csr_array[np.int64, tuple[int, int]], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyIntDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: complex128-like (positional) + def __init__( + self: csr_array[np.complex128, tuple[int]], + /, + arg1: ToShape1D, + shape: ToShape1D | None, + dtype: onp.AnyComplex128DType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: complex128-like (keyword) + def __init__( + self: csr_array[np.complex128, tuple[int]], + /, + arg1: ToShape1D, + shape: ToShape1D | None = None, + *, + dtype: onp.AnyComplex128DType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: complex128-like (positional) + def __init__( + self: csr_array[np.complex128, tuple[int, int]], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyComplex128DType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: complex128-like (keyword) + def __init__( + self: csr_array[np.complex128, tuple[int, int]], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyComplex128DType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d array-like bool, dtype: bool-like | None def __init__( self: csr_array[np.bool_, tuple[int]], /, @@ -153,7 +285,7 @@ class csr_array(_csr_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like bool, dtype: type[bool] | None + @overload # 2-d array-like bool, dtype: bool-like | None def __init__( self: csr_array[np.bool_, tuple[int, int]], /, @@ -164,7 +296,7 @@ class csr_array(_csr_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 1-d array-like ~int, dtype: type[int] | None + @overload # 1-d array-like ~int, dtype: int-like | None def __init__( self: csr_array[np.int64, tuple[int]], /, @@ -175,7 +307,7 @@ class csr_array(_csr_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~int, dtype: type[int] | None + @overload # 2-d array-like ~int, dtype: int-like | None def __init__( self: csr_array[np.int64, tuple[int, int]], /, @@ -186,7 +318,7 @@ class csr_array(_csr_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 1-d array-like ~float, dtype: type[float] | None + @overload # 1-d array-like ~float, dtype: float64-like | None def __init__( self: csr_array[np.float64, tuple[int]], /, @@ -197,7 +329,7 @@ class csr_array(_csr_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~float, dtype: type[float] | None + @overload # 2-d array-like ~float, dtype: float64-like | None def __init__( self: csr_array[np.float64, tuple[int, int]], /, @@ -208,7 +340,7 @@ class csr_array(_csr_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 1-d array-like ~complex, dtype: type[complex] | None + @overload # 1-d array-like ~complex, dtype: complex128-like | None def __init__( self: csr_array[np.complex128, tuple[int]], /, @@ -219,7 +351,7 @@ class csr_array(_csr_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~complex, dtype: type[complex] | None + @overload # 2-d array-like ~complex, dtype: complex128-like | None def __init__( self: csr_array[np.complex128, tuple[int, int]], /, @@ -336,7 +468,7 @@ class csr_matrix(_csr_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like bool, dtype: type[bool] | None + @overload # 2-d array-like bool, dtype: bool-like | None def __init__( self: csr_matrix[np.bool_], /, @@ -347,7 +479,7 @@ class csr_matrix(_csr_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~int, dtype: type[int] | None + @overload # 2-d array-like ~int, dtype: int-like | None def __init__( self: csr_matrix[np.int64], /, @@ -358,7 +490,7 @@ class csr_matrix(_csr_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~float, dtype: type[float] | None + @overload # 2-d array-like ~float, dtype: float64-like | None def __init__( self: csr_matrix[np.float64], /, @@ -369,7 +501,7 @@ class csr_matrix(_csr_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~complex, dtype: type[complex] | None + @overload # 2-d array-like ~complex, dtype: complex128-like | None def __init__( self: csr_matrix[np.complex128], /, diff --git a/scipy-stubs/sparse/_dia.pyi b/scipy-stubs/sparse/_dia.pyi index 6bc45968..1fbadd27 100644 --- a/scipy-stubs/sparse/_dia.pyi +++ b/scipy-stubs/sparse/_dia.pyi @@ -82,18 +82,84 @@ class dia_array(_dia_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d shape-like, dtype: None + @overload # 2-d shape-like, dtype: float64-like | None def __init__( self: dia_array[np.float64], /, arg1: ToShape2D, - shape: None = None, - dtype: None = None, + shape: ToShape2D | None = None, + dtype: onp.AnyFloat64DType | None = None, copy: bool = False, *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.bool, dtype: type[bool] | None + @overload # 2-d shape-like, dtype: bool-like (positional) + def __init__( + self: dia_array[np.bool_], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyBoolDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: bool-like (keyword) + def __init__( + self: dia_array[np.bool_], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyBoolDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: int-like (positional) + def __init__( + self: dia_array[np.int64], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyIntDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: int-like (keyword) + def __init__( + self: dia_array[np.int64], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyIntDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: complex128-like (positional) + def __init__( + self: dia_array[np.complex128], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyComplex128DType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: complex128-like (keyword) + def __init__( + self: dia_array[np.complex128], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyComplex128DType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # matrix-like builtins.bool, dtype: bool-like | None def __init__( self: dia_array[np.bool_], /, @@ -104,7 +170,7 @@ class dia_array(_dia_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.int, dtype: type[int] | None + @overload # matrix-like builtins.int, dtype: int-like | None def __init__( self: dia_array[np.int_], /, @@ -115,7 +181,7 @@ class dia_array(_dia_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.float, dtype: type[float] | None + @overload # matrix-like builtins.float, dtype: float64-like | None def __init__( self: dia_array[np.float64], /, @@ -126,7 +192,7 @@ class dia_array(_dia_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.complex, dtype: type[complex] | None + @overload # matrix-like builtins.complex, dtype: complex128-like | None def __init__( self: dia_array[np.complex128], /, @@ -199,7 +265,7 @@ class dia_matrix(_dia_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.bool, dtype: type[bool] | None + @overload # matrix-like builtins.bool, dtype: bool-like | None def __init__( self: dia_matrix[np.bool_], /, @@ -210,7 +276,7 @@ class dia_matrix(_dia_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.int, dtype: type[int] | None + @overload # matrix-like builtins.int, dtype: int-like | None def __init__( self: dia_matrix[np.int_], /, @@ -221,7 +287,7 @@ class dia_matrix(_dia_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.float, dtype: type[float] | None + @overload # matrix-like builtins.float, dtype: float64-like | None def __init__( self: dia_matrix[np.float64], /, @@ -232,7 +298,7 @@ class dia_matrix(_dia_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.complex, dtype: type[complex] | None + @overload # matrix-like builtins.complex, dtype: complex128-like | None def __init__( self: dia_matrix[np.complex128], /, diff --git a/scipy-stubs/sparse/_dok.pyi b/scipy-stubs/sparse/_dok.pyi index 1a703c7b..cf2ebd89 100644 --- a/scipy-stubs/sparse/_dok.pyi +++ b/scipy-stubs/sparse/_dok.pyi @@ -212,7 +212,7 @@ class dok_array(_dok_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 1-d shape-like, dtype: None + @overload # 1-d shape-like, dtype: float64-like | None def __init__( self: dok_array[np.float64, _1D], /, @@ -223,7 +223,7 @@ class dok_array(_dok_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d shape-like, dtype: None + @overload # 2-d shape-like, dtype: float64-like | None def __init__( self: dok_array[np.float64, _2D], /, @@ -234,7 +234,139 @@ class dok_array(_dok_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 1-d array-like bool, dtype: type[bool] | None + @overload # 1-d shape-like, dtype: bool-like (positional) + def __init__( + self: dok_array[np.bool_, _1D], + /, + arg1: ToShape1D, + shape: ToShape1D | None, + dtype: onp.AnyBoolDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: bool-like (keyword) + def __init__( + self: dok_array[np.bool_, _1D], + /, + arg1: ToShape1D, + shape: ToShape1D | None = None, + *, + dtype: onp.AnyBoolDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: bool-like (positional) + def __init__( + self: dok_array[np.bool_, _2D], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyBoolDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: bool-like (keyword) + def __init__( + self: dok_array[np.bool_, _2D], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyBoolDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: int-like (positional) + def __init__( + self: dok_array[np.int64, _1D], + /, + arg1: ToShape1D, + shape: ToShape1D | None, + dtype: onp.AnyIntDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: int-like (keyword) + def __init__( + self: dok_array[np.int64, _1D], + /, + arg1: ToShape1D, + shape: ToShape1D | None = None, + *, + dtype: onp.AnyIntDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: int-like (positional) + def __init__( + self: dok_array[np.int64, _2D], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyIntDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: int-like (keyword) + def __init__( + self: dok_array[np.int64, _2D], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyIntDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: complex128-like (positional) + def __init__( + self: dok_array[np.complex128, _1D], + /, + arg1: ToShape1D, + shape: ToShape1D | None, + dtype: onp.AnyComplex128DType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d shape-like, dtype: complex128-like (keyword) + def __init__( + self: dok_array[np.complex128, _1D], + /, + arg1: ToShape1D, + shape: ToShape1D | None = None, + *, + dtype: onp.AnyComplex128DType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: complex128-like (positional) + def __init__( + self: dok_array[np.complex128, _2D], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyComplex128DType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: complex128-like (keyword) + def __init__( + self: dok_array[np.complex128, _2D], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyComplex128DType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 1-d array-like bool, dtype: bool-like | None def __init__( self: dok_array[np.bool_, _1D], /, @@ -245,7 +377,7 @@ class dok_array(_dok_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like bool, dtype: type[bool] | None + @overload # 2-d array-like bool, dtype: bool-like | None def __init__( self: dok_array[np.bool_, _2D], /, @@ -256,7 +388,7 @@ class dok_array(_dok_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 1-d array-like ~int, dtype: type[int] | None + @overload # 1-d array-like ~int, dtype: int-like | None def __init__( self: dok_array[np.int64, _1D], /, @@ -267,7 +399,7 @@ class dok_array(_dok_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~int, dtype: type[int] | None + @overload # 2-d array-like ~int, dtype: int-like | None def __init__( self: dok_array[np.int64, _2D], /, @@ -278,7 +410,7 @@ class dok_array(_dok_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 1-d array-like ~float, dtype: type[float] | None + @overload # 1-d array-like ~float, dtype: float64-like | None def __init__( self: dok_array[np.float64, _1D], /, @@ -289,7 +421,7 @@ class dok_array(_dok_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~float, dtype: type[float] | None + @overload # 2-d array-like ~float, dtype: float64-like | None def __init__( self: dok_array[np.float64, _2D], /, @@ -300,7 +432,7 @@ class dok_array(_dok_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 1-d array-like ~complex, dtype: type[complex] | None + @overload # 1-d array-like ~complex, dtype: complex128-like | None def __init__( self: dok_array[np.complex128, _1D], /, @@ -311,7 +443,7 @@ class dok_array(_dok_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d array-like ~complex, dtype: type[complex] | None + @overload # 2-d array-like ~complex, dtype: complex128-like | None def __init__( self: dok_array[np.complex128, _2D], /, @@ -491,7 +623,7 @@ class dok_matrix(_dok_base[_ScalarT_co, _2D], spmatrix[_ScalarT_co], Generic[_Sc *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.bool, dtype: type[bool] | None + @overload # matrix-like builtins.bool, dtype: bool-like | None def __init__( self: dok_matrix[np.bool_], /, @@ -502,7 +634,7 @@ class dok_matrix(_dok_base[_ScalarT_co, _2D], spmatrix[_ScalarT_co], Generic[_Sc *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.int, dtype: type[int] | None + @overload # matrix-like builtins.int, dtype: int-like | None def __init__( self: dok_matrix[np.int64], /, @@ -513,7 +645,7 @@ class dok_matrix(_dok_base[_ScalarT_co, _2D], spmatrix[_ScalarT_co], Generic[_Sc *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.float, dtype: type[float] | None + @overload # matrix-like builtins.float, dtype: float64-like | None def __init__( self: dok_matrix[np.float64], /, @@ -524,7 +656,7 @@ class dok_matrix(_dok_base[_ScalarT_co, _2D], spmatrix[_ScalarT_co], Generic[_Sc *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.complex, dtype: type[complex] | None + @overload # matrix-like builtins.complex, dtype: complex128-like | None def __init__( self: dok_matrix[np.complex128], /, diff --git a/scipy-stubs/sparse/_lil.pyi b/scipy-stubs/sparse/_lil.pyi index 671f79df..c4344512 100644 --- a/scipy-stubs/sparse/_lil.pyi +++ b/scipy-stubs/sparse/_lil.pyi @@ -103,18 +103,84 @@ class lil_array(_lil_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # 2-d shape-like, dtype: None + @overload # 2-d shape-like, dtype: float64-like | None def __init__( self: lil_array[np.float64], /, arg1: ToShape2D, - shape: None = None, + shape: ToShape2D | None = None, dtype: onp.AnyFloat64DType | None = None, copy: bool = False, *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.bool, dtype: type[bool] | None + @overload # 2-d shape-like, dtype: bool-like (positional) + def __init__( + self: lil_array[np.bool_], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyBoolDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: bool-like (keyword) + def __init__( + self: lil_array[np.bool_], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyBoolDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: int-like (positional) + def __init__( + self: lil_array[np.int64], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyIntDType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: int-like (keyword) + def __init__( + self: lil_array[np.int64], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyIntDType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: complex128-like (positional) + def __init__( + self: lil_array[np.complex128], + /, + arg1: ToShape2D, + shape: ToShape2D | None, + dtype: onp.AnyComplex128DType, + copy: bool = False, + *, + maxprint: int | None = None, + ) -> None: ... + @overload # 2-d shape-like, dtype: complex128-like (keyword) + def __init__( + self: lil_array[np.complex128], + /, + arg1: ToShape2D, + shape: ToShape2D | None = None, + *, + dtype: onp.AnyComplex128DType, + copy: bool = False, + maxprint: int | None = None, + ) -> None: ... + @overload # matrix-like builtins.bool, dtype: bool-like | None def __init__( self: lil_array[np.bool_], /, @@ -125,7 +191,7 @@ class lil_array(_lil_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.int, dtype: type[int] | None + @overload # matrix-like builtins.int, dtype: int-like | None def __init__( self: lil_array[np.int_], /, @@ -136,7 +202,7 @@ class lil_array(_lil_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.float, dtype: type[float] | None + @overload # matrix-like builtins.float, dtype: float64-like | None def __init__( self: lil_array[np.float64], /, @@ -147,7 +213,7 @@ class lil_array(_lil_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.complex, dtype: type[complex] | None + @overload # matrix-like builtins.complex, dtype: complex128-like | None def __init__( self: lil_array[np.complex128], /, @@ -229,7 +295,7 @@ class lil_matrix(_lil_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.bool, dtype: type[bool] | None + @overload # matrix-like builtins.bool, dtype: bool-like | None def __init__( self: lil_matrix[np.bool_], /, @@ -240,7 +306,7 @@ class lil_matrix(_lil_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.int, dtype: type[int] | None + @overload # matrix-like builtins.int, dtype: int-like | None def __init__( self: lil_matrix[np.int_], /, @@ -251,7 +317,7 @@ class lil_matrix(_lil_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.float, dtype: type[float] | None + @overload # matrix-like builtins.float, dtype: float64-like | None def __init__( self: lil_matrix[np.float64], /, @@ -262,7 +328,7 @@ class lil_matrix(_lil_base[_ScalarT_co], spmatrix[_ScalarT_co], Generic[_ScalarT *, maxprint: int | None = None, ) -> None: ... - @overload # matrix-like builtins.complex, dtype: type[complex] | None + @overload # matrix-like builtins.complex, dtype: complex128-like | None def __init__( self: lil_matrix[np.complex128], /, diff --git a/tests/sparse/test_csr.pyi b/tests/sparse/test_csr.pyi index 9e38a930..6e6f8130 100644 --- a/tests/sparse/test_csr.pyi +++ b/tests/sparse/test_csr.pyi @@ -25,6 +25,40 @@ csr_matrix(1) # type: ignore[call-overload] # pyright: ignore[reportArgumentTy assert_type(csr_array((2,)), csr_array[np.float64, tuple[int]]) assert_type(csr_array((2, 3)), csr_array[np.float64, tuple[int, int]]) +assert_type(csr_array((2,), dtype=np.bool_), csr_array[np.bool_, tuple[int]]) +assert_type(csr_array((2, 3), dtype=np.bool_), csr_array[np.bool_, tuple[int, int]]) +assert_type(csr_array((2,), dtype=bool), csr_array[np.bool_, tuple[int]]) +assert_type(csr_array((2, 3), dtype=bool), csr_array[np.bool_, tuple[int, int]]) +assert_type(csr_array((2,), dtype="bool"), csr_array[np.bool_, tuple[int]]) +assert_type(csr_array((2, 3), dtype="bool"), csr_array[np.bool_, tuple[int, int]]) +assert_type(csr_array((2,), dtype=np.int64), csr_array[np.int64, tuple[int]]) +assert_type(csr_array((2, 3), dtype=np.int64), csr_array[np.int64, tuple[int, int]]) +assert_type(csr_array((2,), dtype=int), csr_array[np.int64, tuple[int]]) +assert_type(csr_array((2, 3), dtype=int), csr_array[np.int64, tuple[int, int]]) +assert_type(csr_array((2,), dtype="int"), csr_array[np.int64, tuple[int]]) +assert_type(csr_array((2, 3), dtype="int"), csr_array[np.int64, tuple[int, int]]) +assert_type(csr_array((2,), dtype=np.float64), csr_array[np.float64, tuple[int]]) +assert_type(csr_array((2, 3), dtype=np.float64), csr_array[np.float64, tuple[int, int]]) +assert_type(csr_array((2,), dtype=float), csr_array[np.float64, tuple[int]]) +assert_type(csr_array((2, 3), dtype=float), csr_array[np.float64, tuple[int, int]]) +assert_type(csr_array((2,), dtype="float"), csr_array[np.float64, tuple[int]]) +assert_type(csr_array((2, 3), dtype="float"), csr_array[np.float64, tuple[int, int]]) +assert_type(csr_array((2,), dtype=np.complex128), csr_array[np.complex128, tuple[int]]) +assert_type(csr_array((2, 3), dtype=np.complex128), csr_array[np.complex128, tuple[int, int]]) +assert_type(csr_array((2,), dtype=complex), csr_array[np.complex128, tuple[int]]) +assert_type(csr_array((2, 3), dtype=complex), csr_array[np.complex128, tuple[int, int]]) +assert_type(csr_array((2,), dtype="complex"), csr_array[np.complex128, tuple[int]]) +assert_type(csr_array((2, 3), dtype="complex"), csr_array[np.complex128, tuple[int, int]]) +assert_type(csr_array((2,), None, None), csr_array[np.float64, tuple[int]]) +assert_type(csr_array((2, 3), None, None), csr_array[np.float64, tuple[int, int]]) +assert_type(csr_array((2,), None, np.bool_), csr_array[np.bool_, tuple[int]]) +assert_type(csr_array((2, 3), None, np.bool_), csr_array[np.bool_, tuple[int, int]]) +assert_type(csr_array((2,), None, np.int64), csr_array[np.int64, tuple[int]]) +assert_type(csr_array((2, 3), None, np.int64), csr_array[np.int64, tuple[int, int]]) +assert_type(csr_array((2,), None, np.float64), csr_array[np.float64, tuple[int]]) +assert_type(csr_array((2, 3), None, np.float64), csr_array[np.float64, tuple[int, int]]) +assert_type(csr_array((2,), None, np.complex128), csr_array[np.complex128, tuple[int]]) +assert_type(csr_array((2, 3), None, np.complex128), csr_array[np.complex128, tuple[int, int]]) assert_type(csr_matrix((2, 3)), csr_matrix[np.float64]) assert_type(csr_array(seq_bool), csr_array[np.bool_, tuple[int]])