Skip to content

Commit 502a2d9

Browse files
committed
MAINT: enum
1 parent 9ee6341 commit 502a2d9

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

array_api_tests/test_dlpack.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def test_dlpack_device(dtype, data):
2828
x = xp.empty(3, dtype=dtype)
2929
device_type, device_id = x.__dlpack_device__()
3030

31-
assert device_type in DLPackDeviceEnum
31+
assert DLPackDeviceEnum(int(device_type))
3232
assert isinstance(device_id, int)
3333

3434

0 commit comments

Comments
 (0)