🤡 pretend that ndarray and generic implement __buffer__ before py312

Author: jorenham

src/numpy-stubs/__init__.pyi

@@ -23,10 +23,9 @@ from typing import (
     TypedDict,
     final,
     overload,
-    runtime_checkable,
     type_check_only,
 )
-from typing_extensions import CapsuleType, LiteralString, Never, Protocol, Self, TypeVar, Unpack, deprecated, override
+from typing_extensions import Buffer, CapsuleType, LiteralString, Never, Protocol, Self, TypeVar, Unpack, deprecated, override
 
 from . import (
     __config__ as __config__,
@@ -463,22 +462,6 @@ from .lib._utils_impl import get_include, info, show_runtime
 from .matrixlib import asmatrix, bmat, matrix
 from .version import __version__
 
-@runtime_checkable
-class _Buffer(Protocol):
-    def __buffer__(self, flags: int, /) -> memoryview: ...
-
-if sys.version_info >= (3, 12):
-    _SupportsBuffer: TypeAlias = _Buffer
-else:
-    import array as _array
-    import mmap as _mmap
-
-    from numpy import distutils as distutils  # noqa: ICN003
-
-    _SupportsBuffer: TypeAlias = (
-        _Buffer | bytes | bytearray | memoryview | _array.array[Any] | _mmap.mmap | NDArray[Any] | generic
-    )
-
 __all__ = [  # noqa: RUF022
     # __numpy_submodules__
     "char", "core", "ctypeslib", "dtypes", "exceptions", "f2py", "fft", "lib", "linalg", "ma", "polynomial", "random", "rec",
@@ -1993,7 +1976,7 @@ class ndarray(_ArrayOrScalarCommon, Generic[_ShapeT_co, _DTypeT_co]):
         cls,
         shape: _ShapeLike,
         dtype: DTypeLike = float,  # noqa: PYI011
-        buffer: _SupportsBuffer | None = None,
+        buffer: Buffer | None = None,
         offset: CanIndex = 0,
         strides: _ShapeLike | None = None,
         order: _OrderKACF | None = None,
@@ -3836,9 +3819,6 @@ class generic(_ArrayOrScalarCommon, Generic[_ItemT_co]):
     @abc.abstractmethod
     def __init__(self, /, *args: Any, **kwargs: Any) -> None: ...
 
-    if sys.version_info >= (3, 12):
-        def __buffer__(self, flags: int, /) -> memoryview: ...
-
     #
     @overload
     def __array__(self, dtype: None = None, /) -> ndarray[tuple[()], dtype[Self]]: ...

src/numpy-stubs/_core/_multiarray_umath.pyi

@@ -19,7 +19,7 @@ from typing import (
     overload,
     type_check_only,
 )
-from typing_extensions import CapsuleType, Self, TypeAliasType, TypeVar, Unpack, deprecated
+from typing_extensions import Buffer, CapsuleType, Self, TypeAliasType, TypeVar, Unpack, deprecated
 
 import numpy as np
 import numpy.typing as npt
@@ -30,7 +30,6 @@ from numpy import (  # noqa: ICN003
     _ModeKind,
     _OrderCF,
     _OrderKACF,
-    _SupportsBuffer,
     # NOTE: These implicitly re-exported ufuncs are defined in this ext-module at runtime
     absolute as absolute,
     add as add,
@@ -1165,23 +1164,23 @@ def fromiter(
 #
 @overload
 def frombuffer(
-    buffer: _SupportsBuffer,
+    buffer: Buffer,
     *,
     count: SupportsIndex = -1,
     offset: SupportsIndex = 0,
     **kwargs: Unpack[_KwargsL],
 ) -> _Array[np.float64]: ...
 @overload
 def frombuffer(
-    buffer: _SupportsBuffer,
+    buffer: Buffer,
     dtype: _DTypeLike[_ScalarT],
     count: SupportsIndex = -1,
     offset: SupportsIndex = 0,
     **kwargs: Unpack[_KwargsL],
 ) -> _Array[_ScalarT]: ...
 @overload
 def frombuffer(
-    buffer: _SupportsBuffer,
+    buffer: Buffer,
     dtype: npt.DTypeLike,
     count: SupportsIndex = -1,
     offset: SupportsIndex = 0,

src/numpy-stubs/_core/defchararray.pyi

@@ -1,9 +1,9 @@
 from _typeshed import ConvertibleToInt
 from typing import Any, Literal as L, SupportsIndex, TypeAlias, overload
-from typing_extensions import Never, Self, TypeVar, override
+from typing_extensions import Buffer, Never, Self, TypeVar, override
 
 import numpy as np
-from numpy import _OrderKACF as _Order, _SortKind, _SupportsBuffer  # noqa: ICN003
+from numpy import _OrderKACF as _Order, _SortKind  # noqa: ICN003
 from numpy._typing import (
     NDArray,
     _ArrayLikeAnyString_co as _ToAnyCharND,
@@ -158,7 +158,7 @@ class chararray(np.ndarray[_ShapeT_co, _DTypeT_co]):
         shape: _ToShape,
         itemsize: ConvertibleToInt = 1,
         unicode: L[False] = False,
-        buffer: _SupportsBuffer | None = None,
+        buffer: Buffer | None = None,
         offset: SupportsIndex = 0,
         strides: _ToShape | None = None,
         order: _Order = "C",
@@ -169,7 +169,7 @@ class chararray(np.ndarray[_ShapeT_co, _DTypeT_co]):
         shape: _ToShape,
         itemsize: ConvertibleToInt,
         unicode: L[True],
-        buffer: _SupportsBuffer | None = None,
+        buffer: Buffer | None = None,
         offset: SupportsIndex = 0,
         strides: _ToShape | None = None,
         order: _Order = "C",
@@ -181,7 +181,7 @@ class chararray(np.ndarray[_ShapeT_co, _DTypeT_co]):
         itemsize: ConvertibleToInt = 1,
         *,
         unicode: L[True],
-        buffer: _SupportsBuffer | None = None,
+        buffer: Buffer | None = None,
         offset: SupportsIndex = 0,
         strides: _ToShape | None = None,
         order: _Order = "C",

src/numpy-stubs/_core/records.pyi

@@ -1,10 +1,10 @@
 from _typeshed import StrOrBytesPath
 from collections.abc import Iterable, Sequence
 from typing import IO, Any, ClassVar, Literal as L, Protocol, SupportsIndex, TypeAlias, overload, type_check_only
-from typing_extensions import TypeVar
+from typing_extensions import Buffer, TypeVar
 
 import numpy as np
-from numpy import _ByteOrder, _OrderKACF, _SupportsBuffer  # noqa: ICN003
+from numpy import _ByteOrder, _OrderKACF  # noqa: ICN003
 from numpy._typing import ArrayLike, DTypeLike, NDArray, _ArrayLikeVoid_co, _NestedSequence, _ShapeLike
 
 __all__ = [
@@ -66,7 +66,7 @@ class recarray(np.ndarray[_ShapeT_co, _DTypeT_co]):
         subtype,
         shape: _ShapeLike,
         dtype: None = None,
-        buf: _SupportsBuffer | None = None,
+        buf: Buffer | None = None,
         offset: SupportsIndex = 0,
         strides: _ShapeLike | None = None,
         *,
@@ -82,7 +82,7 @@ class recarray(np.ndarray[_ShapeT_co, _DTypeT_co]):
         subtype,
         shape: _ShapeLike,
         dtype: DTypeLike,
-        buf: _SupportsBuffer | None = None,
+        buf: Buffer | None = None,
         offset: SupportsIndex = ...,
         strides: _ShapeLike | None = None,
         formats: None = None,
@@ -158,7 +158,7 @@ def fromrecords(
 # exported in `numpy.rec`
 @overload
 def fromstring(
-    datastring: _SupportsBuffer,
+    datastring: Buffer,
     dtype: DTypeLike,
     shape: _ShapeLike | None = None,
     offset: int = 0,
@@ -170,7 +170,7 @@ def fromstring(
 ) -> _RecArray[record]: ...
 @overload
 def fromstring(
-    datastring: _SupportsBuffer,
+    datastring: Buffer,
     dtype: None = None,
     shape: _ShapeLike | None = None,
     offset: int = 0,
@@ -214,7 +214,7 @@ def fromfile(
 
 # exported in `numpy.rec`
 @overload
-def array(
+def array(  # type: ignore[overload-overlap]
     obj: _SCT | NDArray[_SCT],
     dtype: None = None,
     shape: _ShapeLike | None = None,