Skip to content

GitLab

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

Fix lint (#7408)

parent 82cf540b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/test_image.py
View file @ ed9207d7
...@@ -368,10 +368,10 @@ def test_decode_jpeg_cuda(mode, img_path, scripted): ...@@ -368,10 +368,10 @@ def test_decode_jpeg_cuda(mode, img_path, scripted):
# Some difference expected between jpeg implementations # Some difference expected between jpeg implementations
assert (img.float() - img_nvjpeg.cpu().float()).abs().mean() < 2 assert (img.float() - img_nvjpeg.cpu().float()).abs().mean() < 2
@needs_cuda @needs_cuda
def test_decode_image_cuda_raises(): def test_decode_image_cuda_raises():
data = torch.randint(0, 127, size=(255,), device="cuda", dtype=torch.uint8) data = torch.randint(0, 127, size=(255,), device="cuda", dtype=torch.uint8)
exception_raised = True
with pytest.raises(RuntimeError): with pytest.raises(RuntimeError):
decode_image(data) 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