TYP: Use collections.abc.Buffer in the Array API

numpy · Sep 14, 2023 · 56e580d · 56e580d
1 parent 5ef53c6
commit 56e580d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/numpy/array_api/_typing.py b/numpy/array_api/_typing.py
@@ -17,6 +17,8 @@
     "PyCapsule",
 ]
 
+import sys
+
 from typing import (
     Any,
     Literal,
@@ -63,8 +65,11 @@ def __len__(self, /) -> int: ...
     float64,
 ]]
 
+if sys.version_info >= (3, 12):
+    from collections.abc import Buffer as SupportsBufferProtocol
+else:
+    SupportsBufferProtocol = Any
 
-SupportsBufferProtocol = Any
 PyCapsule = Any
 
 class SupportsDLPack(Protocol):