Fix passing of keyword arguments in __dlpack__ for NumPy 2.1

Author: asmeurer

array_api_strict/_array_object.py

@@ -595,7 +595,15 @@ def __dlpack__(
                 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)
+        else:
+            kwargs = {'stream': stream}
+            if max_version is not _default:
+                kwargs['max_version'] = max_version
+            if dl_device is not _default:
+                kwargs['dl_device'] = dl_device
+            if copy is not _default:
+                kwargs['copy'] = copy
+            return self._array.__dlpack__(**kwargs)
 
     def __dlpack_device__(self: Array, /) -> Tuple[IntEnum, int]:
         """

array_api_strict/tests/test_array_object.py

@@ -460,18 +460,27 @@ def dlpack_2023_12(api_version):
     a.__dlpack__()
 
 
-    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:
-                  a.__dlpack__(dl_device=None))
-    pytest.raises(exception, lambda:
-                  a.__dlpack__(max_version=(1, 0)))
-    pytest.raises(exception, lambda:
-                  a.__dlpack__(max_version=None))
-    pytest.raises(exception, lambda:
-                    a.__dlpack__(copy=False))
-    pytest.raises(exception, lambda:
-                    a.__dlpack__(copy=True))
-    pytest.raises(exception, lambda:
-                    a.__dlpack__(copy=None))
+    if np.__version__ < '2.1':
+        exception = NotImplementedError if api_version >= '2023.12' else ValueError
+        pytest.raises(exception, lambda:
+                      a.__dlpack__(dl_device=CPU_DEVICE))
+        pytest.raises(exception, lambda:
+                      a.__dlpack__(dl_device=None))
+        pytest.raises(exception, lambda:
+                      a.__dlpack__(max_version=(1, 0)))
+        pytest.raises(exception, lambda:
+                      a.__dlpack__(max_version=None))
+        pytest.raises(exception, lambda:
+                        a.__dlpack__(copy=False))
+        pytest.raises(exception, lambda:
+                        a.__dlpack__(copy=True))
+        pytest.raises(exception, lambda:
+                        a.__dlpack__(copy=None))
+    else:
+        a.__dlpack__(dl_device=CPU_DEVICE)
+        a.__dlpack__(dl_device=None)
+        a.__dlpack__(max_version=(1, 0))
+        a.__dlpack__(max_version=None)
+        a.__dlpack__(copy=False)
+        a.__dlpack__(copy=True)
+        a.__dlpack__(copy=None)