Skip to content

Commit c8b12fd

Browse files
authored
Avoid cmyk in nvjpeg tests (#4246)
1 parent 3b8d51e commit c8b12fd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

test/test_image.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -326,7 +326,8 @@ def test_decode_jpeg_cuda(mode, img_path, scripted):
326326
@pytest.mark.parametrize('cuda_device', ('cuda', 'cuda:0', torch.device('cuda')))
327327
def test_decode_jpeg_cuda_device_param(cuda_device):
328328
"""Make sure we can pass a string or a torch.device as device param"""
329-
data = read_file(next(get_images(IMAGE_ROOT, ".jpg")))
329+
path = next(path for path in get_images(IMAGE_ROOT, ".jpg") if 'cmyk' not in path)
330+
data = read_file(path)
330331
decode_jpeg(data, device=cuda_device)
331332

332333

0 commit comments

Comments
 (0)