Put back more stuff

Author: NicolasHug

test/test_datasets_video_utils_opt.py

@@ -15,12 +15,13 @@
     write_jpeg,
     write_png,
 )
-from .video import write_video, read_video
+from .video import write_video, read_video, read_video_timestamps
 
 
 __all__ = [
     "write_video",
     "read_video",
+    "read_video_timestamps",
     "ImageReadMode",
     "decode_image",
     "decode_jpeg",

torchvision/io/__init__.py

@@ -44,6 +44,8 @@ def _check_av_available() -> None:
     if isinstance(av, Exception):
         raise av
 
+def _av_available() -> bool:
+    return not isinstance(av, Exception)
 
 def write_video(
     filename: str,