diff --git a/torchvision/io/__init__.py b/torchvision/io/__init__.py index 36be33884b8..13958fb9ab4 100644 --- a/torchvision/io/__init__.py +++ b/torchvision/io/__init__.py @@ -1,8 +1,16 @@ from .video import write_video, read_video, read_video_timestamps -from ._video_opt import _read_video_from_file, _read_video_timestamps_from_file, _HAS_VIDEO_OPT +from ._video_opt import ( + _read_video_from_file, + _read_video_timestamps_from_file, + _read_video_from_memory, + _read_video_timestamps_from_memory, + _HAS_VIDEO_OPT, +) __all__ = [ 'write_video', 'read_video', 'read_video_timestamps', - '_read_video_from_file', '_read_video_timestamps_from_file', '_HAS_VIDEO_OPT', + '_read_video_from_file', '_read_video_timestamps_from_file', + '_read_video_from_memory', '_read_video_timestamps_from_memory', + '_HAS_VIDEO_OPT', ]