A little more refactoring

Author: atalman

test/smoke_test/smoke_test.py

@@ -41,18 +41,18 @@ 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 decode_image, read_file
-    img_jpg = read_file(str(pathlib.Path(__file__).parent / 'assets' / 'rgb_pytorch.jpg'))
-    img_jpg_nv = decode_image(img_jpg)
-    img_png = read_file(str(pathlib.Path(__file__).parent  / 'assets' / 'rgb_pytorch.png'))
-    img_png_nv = decode_image(img_png)
+    from torchvision.io import read_image
+    img_jpg = read_image(str(pathlib.Path(__file__).parent / 'assets' / 'rgb_pytorch.jpg'))
+    assert img_jpg.ndim == 3 and img_jpg.numel() > 100
+    img_png = read_image(str(pathlib.Path(__file__).parent  / 'assets' / 'rgb_pytorch.png'))
+    assert img_png.ndim == 3 and img_png.numel() > 100
 
 def smoke_test_torchvision_resnet50_classify() -> None:
     from torchvision.io import read_image
     from torchvision.models import resnet50, ResNet50_Weights
 
     img = read_image(str(pathlib.Path(__file__).parent / 'assets' / 'dog2.jpg'))
-    assert img.ndim == 3 and img.numel() > 100
+
     # Step 1: Initialize model with the best available weights
     weights = ResNet50_Weights.DEFAULT
     model = resnet50(weights=weights)