Unverified Commit c8b12fdb authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Avoid cmyk in nvjpeg tests (#4246)

parent 3b8d51e7
...@@ -326,7 +326,8 @@ def test_decode_jpeg_cuda(mode, img_path, scripted): ...@@ -326,7 +326,8 @@ def test_decode_jpeg_cuda(mode, img_path, scripted):
@pytest.mark.parametrize('cuda_device', ('cuda', 'cuda:0', torch.device('cuda'))) @pytest.mark.parametrize('cuda_device', ('cuda', 'cuda:0', torch.device('cuda')))
def test_decode_jpeg_cuda_device_param(cuda_device): def test_decode_jpeg_cuda_device_param(cuda_device):
"""Make sure we can pass a string or a torch.device as device param""" """Make sure we can pass a string or a torch.device as device param"""
data = read_file(next(get_images(IMAGE_ROOT, ".jpg"))) path = next(path for path in get_images(IMAGE_ROOT, ".jpg") if 'cmyk' not in path)
data = read_file(path)
decode_jpeg(data, device=cuda_device) decode_jpeg(data, device=cuda_device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment