Unverified Commit 42e7f1f0 authored by James Thewlis's avatar James Thewlis Committed by GitHub
Browse files

Warn if torchvision imported from repo root (#2759)

* Warn if torchvision imported from repo root

This is likely to fail as mentioned in #2239

* Only warn if ops are missing
parent 039c9288
import warnings
import os
from .extension import _HAS_OPS
......@@ -16,6 +17,14 @@ try:
except ImportError:
pass
# Check if torchvision is being imported within the root folder
if (not _HAS_OPS and os.path.dirname(os.path.realpath(__file__)) ==
os.path.join(os.path.realpath(os.getcwd()), 'torchvision')):
message = ('You are importing torchvision within its own root folder ({}). '
'This is not expected to work and may give errors. Please exit the '
'torchvision project source and relaunch your python interpreter.')
warnings.warn(message.format(os.getcwd()))
_image_backend = 'PIL'
_video_backend = "pyav"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment