Skip to content

GitLab

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

More robust check in tests for 16 bits images (#5652)

parent fc63f828
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/test_image.py
View file @ 90dace8d
...@@ -158,7 +158,7 @@ def test_decode_png(img_path, pil_mode, mode): ...@@ -158,7 +158,7 @@ def test_decode_png(img_path, pil_mode, mode):
img_pil = normalize_dimensions(img_pil) img_pil = normalize_dimensions(img_pil)
if "16" in img_path: if img_path.endswith("16.png"):
# 16 bits image decoding is supported, but only as a private API # 16 bits image decoding is supported, but only as a private API
# FIXME: see https://github.com/pytorch/vision/issues/4731 for potential solutions to making it public # FIXME: see https://github.com/pytorch/vision/issues/4731 for potential solutions to making it public
with pytest.raises(RuntimeError, match="At most 8-bit PNG images are supported"): with pytest.raises(RuntimeError, match="At most 8-bit PNG images are supported"):
......
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