fix missing torchdata error message (#5738)

d0c92dc9 · Philip Meier · GitHub · c6c9ab98 · d0c92dc9
Unverified Commit d0c92dc9 authored Apr 05, 2022 by Philip Meier Committed by GitHub Apr 05, 2022
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 4 deletions

torchvision/prototype/datasets/__init__.py torchvision/prototype/datasets/__init__.py +3 -4

No files found.
--- a/torchvision/prototype/datasets/__init__.py
+++ b/torchvision/prototype/datasets/__init__.py
 try:
    import torchdata
-except (ModuleNotFoundError, TypeError) as error:
+except ModuleNotFoundError:
    raise ModuleNotFoundError(
        "`torchvision.prototype.datasets` depends on PyTorch's `torchdata` (https://github.com/pytorch/data). "
-        "You can install it with `pip install git+https://github.com/pytorch/data.git`. "
+        "You can install it with `pip install --pre torchdata --extra-index-url https://download.pytorch.org/whl/nightly/cpu"
-        "Note that you cannot install it with `pip install torchdata`, since this is another package."
+    ) from None
-    ) from error
 from . import utils
 from ._home import home