fix: swap names for __dlpack__ (#2741)

scikit-hep · Oct 4, 2023 · c377185 · c377185
1 parent db70407
commit c377185
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/awkward/index.py b/src/awkward/index.py
@@ -145,10 +145,10 @@ def __cuda_array_interface__(self):
     def __array_interface__(self):
         return self._data.__array_interface__
 
-    def __dlpack__(self) -> tuple[int, int]:
+    def __dlpack_device__(self) -> tuple[int, int]:
         return self._data.__dlpack_device__()
 
-    def __dlpack_device__(self, stream: Any = None) -> Any:
+    def __dlpack__(self, stream: Any = None) -> Any:
         if stream is None:
             return self._data.__dlpack__()
         else: