adding the warning on unsucessful import

bryant1410 · Oct 8, 2020 · afef7df · afef7df
1 parent 0fc157d
commit afef7df
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torchvision/io/_video_opt.py b/torchvision/io/_video_opt.py
@@ -26,7 +26,9 @@
         torch.ops.load_library(ext_specs.origin)
         _HAS_VIDEO_OPT = True
 except (ImportError, OSError):
-    pass
+    warnings.warn("Could not import video_reader bakcend."
+                  + " Make sure ffmpeg is installed properly"
+                  + " and rebuild torchvision.")
 
 
 default_timebase = Fraction(0, 1)