Testing

Author: atalman

test/smoke_test/smoke_test.py

@@ -41,7 +41,7 @@ def smoke_test_torchvision() -> None:
     print('Is torchvision useable?', all(x is not None for x in [torch.ops.image.decode_png, torch.ops.torchvision.roi_align]))
 
 def smoke_test_torchvision_read_decode() -> None:
-    from torchvision.io import read_file, decode_jpeg, decode_png
+    from torchvision.io import read_file, read_image, decode_jpeg, decode_png
     img_jpg = read_file(str(pathlib.Path(__file__).parent / 'assets' / 'rgb_pytorch.jpg'))
     img_jpg_nv = decode_jpeg(img_jpg)
     img_png = read_image(str(pathlib.Path(__file__).parent  / 'assets' / 'rgb_pytorch.png'))