We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9ee6341 commit 502a2d9Copy full SHA for 502a2d9
1 file changed
array_api_tests/test_dlpack.py
@@ -28,7 +28,7 @@ def test_dlpack_device(dtype, data):
28
x = xp.empty(3, dtype=dtype)
29
device_type, device_id = x.__dlpack_device__()
30
31
- assert device_type in DLPackDeviceEnum
+ assert DLPackDeviceEnum(int(device_type))
32
assert isinstance(device_id, int)
33
34
0 commit comments