enable impors (#2918)

1fe1e110 · So Uchida · GitHub · 75e4a7dd · 1fe1e110
Unverified Commit 1fe1e110 authored Oct 28, 2020 by So Uchida Committed by GitHub Oct 28, 2020
Show whitespace changes
Inline Side-by-side

Showing with 12 additions and 4 deletions

torchvision/io/__init__.py torchvision/io/__init__.py +12 -4

No files found.
--- a/torchvision/io/__init__.py
+++ b/torchvision/io/__init__.py
@@ -17,11 +17,15 @@ from .video import (
    write_video,
 )
 from .image import (
-    read_image,
    decode_image,
+    decode_jpeg,
+    decode_png,
    encode_jpeg,
-    write_jpeg,
    encode_png,
+    read_file,
+    read_image,
+    write_file,
+    write_jpeg,
    write_png,
 )

@@ -164,11 +168,15 @@ __all__ = [
    "_read_video_meta_data",
    "VideoMetaData",
    "Timebase",
-    "read_image",
    "decode_image",
+    "decode_jpeg",
+    "decode_png",
    "encode_jpeg",
-    "write_jpeg",
    "encode_png",
+    "read_file",
+    "read_image",
+    "write_file",
+    "write_jpeg",
    "write_png",
    "Video",
 ]