fix: support dlpack

scikit-hep · Oct 31, 2023 · 41f2c56 · 41f2c56
1 parent 59c5f3e
commit 41f2c56
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/awkward/_nplikes/typetracer.py b/src/awkward/_nplikes/typetracer.py
@@ -495,6 +495,12 @@ def __int__(self) -> int:
     def __index__(self) -> int:
         raise RuntimeError("cannot realise an unknown value")
 
+    def __dlpack_device__(self) -> tuple[int, int]:
+        raise RuntimeError("cannot realise an unknown value")
+
+    def __dlpack__(self, stream: Any = None) -> Any:
+        raise RuntimeError("cannot realise an unknown value")
+
 
 def _scalar_type_of(obj) -> numpy.dtype:
     if is_unknown_scalar(obj):