Skip DLPack CPU export test for incompatible Numpy (#4904)

Signed-off-by: Albert Wolant <[email protected]>
NVIDIA · Jun 12, 2023 · 65bb9bc · 65bb9bc
1 parent d246516
commit 65bb9bc
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/dali/test/python/test_backend_impl.py b/dali/test/python/test_backend_impl.py
@@ -348,6 +348,12 @@ def test_tensor_str_sample():
 
 
 def test_tensor_expose_dlpack_capsule():
+    # TODO(awolant): Numpy versions for Python 3.6 and 3.7 do not
+    # support from_dlpack. When we upgrade DLPack support for DALI
+    # this test needs to be changed.
+    if not hasattr(np, "from_dlpack"):
+        raise SkipTest("Test requires Numpy DLPack support.")
+
     arr = np.arange(20)
     tensor = TensorCPU(arr, "NHWC")