util: Explicitly call __dlpack__ built-in method in `xp2tensorflo…

…w` (#757) `tf.experimental.dlpack.from_dlpack` expects a `PyCapsule` object.
explosion · Sep 7, 2022 · fc323e1 · fc323e1
1 parent fba3bf0
commit fc323e1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/thinc/util.py b/thinc/util.py
@@ -404,7 +404,8 @@ def xp2tensorflow(
         dlpack_tensor = xp_tensor.toDlpack()  # type: ignore
         tf_tensor = tf.experimental.dlpack.from_dlpack(dlpack_tensor)
     elif hasattr(xp_tensor, "__dlpack__"):
-        tf_tensor = tf.experimental.dlpack.from_dlpack(xp_tensor)
+        dlpack_tensor = xp_tensor.__dlpack__()  # type: ignore
+        tf_tensor = tf.experimental.dlpack.from_dlpack(dlpack_tensor)
     else:
         tf_tensor = tf.convert_to_tensor(xp_tensor)
     if as_variable: