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

Fix lint (#7408)

parent 82cf540b
......@@ -368,10 +368,10 @@ def test_decode_jpeg_cuda(mode, img_path, scripted):
# Some difference expected between jpeg implementations
assert (img.float() - img_nvjpeg.cpu().float()).abs().mean() < 2
@needs_cuda
def test_decode_image_cuda_raises():
data = torch.randint(0, 127, size=(255,), device="cuda", dtype=torch.uint8)
exception_raised = True
with pytest.raises(RuntimeError):
decode_image(data)
......
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