Allow __dlpack__ to work with newer versions of NumPy

Author: asmeurer

array_api_strict/_array_object.py

@@ -586,15 +586,16 @@ def __dlpack__(
             if copy is not _default:
                 raise ValueError("The copy argument to __dlpack__ requires at least version 2023.12 of the array API")
 
-        # Going to wait for upstream numpy support
-        if max_version not in [_default, None]:
-            raise NotImplementedError("The max_version argument to __dlpack__ is not yet implemented")
-        if dl_device not in [_default, None]:
-            raise NotImplementedError("The device argument to __dlpack__ is not yet implemented")
-        if copy not in [_default, None]:
-            raise NotImplementedError("The copy argument to __dlpack__ is not yet implemented")
-
-        return self._array.__dlpack__(stream=stream)
+        if np.__version__ < '2.1':
+            if max_version not in [_default, None]:
+                raise NotImplementedError("The max_version argument to __dlpack__ is not yet implemented")
+            if dl_device not in [_default, None]:
+                raise NotImplementedError("The device argument to __dlpack__ is not yet implemented")
+            if copy not in [_default, None]:
+                raise NotImplementedError("The copy argument to __dlpack__ is not yet implemented")
+
+            return self._array.__dlpack__(stream=stream)
+        return self._array.__dlpack__(stream=stream, max_version=max_version, dl_device=dl_device, copy=copy)
 
     def __dlpack_device__(self: Array, /) -> Tuple[IntEnum, int]:
         """

array_api_strict/tests/test_array_object.py

@@ -460,7 +460,7 @@ def dlpack_2023_12(api_version):
     a.__dlpack__()
 
 
-    exception = NotImplementedError if api_version >= '2023.12' else ValueError
+    exception = NotImplementedError if api_version >= '2023.12' and np.__version__ < '2.1' else ValueError
     pytest.raises(exception, lambda:
                   a.__dlpack__(dl_device=CPU_DEVICE))
     pytest.raises(exception, lambda: