filter out the torch "meta" device

Author: ev-br

array_api_tests/test_dlpack.py

@@ -21,6 +21,15 @@ class DLPackDeviceEnum(Enum):
     ONE_API = 14
 
 
+def filter_meta(devices):
+    # hack: filter out torch.device(type="meta")
+    try:
+        import torch
+        return [x for x in devices if x != torch.device(type="meta")]
+    except Exception:
+        return devices
+
+
 @given(dtype=hh.all_dtypes, data=st.data())
 def test_dlpack_device(dtype, data):
     """Test the array object __dlpack_device__ method."""
@@ -39,6 +48,7 @@ def test_dlpack_device(dtype, data):
 )
 def test_from_dlpack(x, copy_kw, data):
     devices = xp.__array_namespace_info__().devices()
+    devices = filter_meta(devices)
     tgt_device_kw = data.draw(
         hh.kwargs(device=st.sampled_from(devices) | st.none())
     )