fix missing torchdata error message (#5738)

pytorch · Apr 5, 2022 · d0c92dc · d0c92dc
1 parent c6c9ab9
commit d0c92dc
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/torchvision/prototype/datasets/__init__.py b/torchvision/prototype/datasets/__init__.py
@@ -1,11 +1,10 @@
 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`. "
- "Note that you cannot install it with `pip install torchdata`, since this is another package."
- ) from error
+ "You can install it with `pip install --pre torchdata --extra-index-url https://download.pytorch.org/whl/nightly/cpu"
+ ) from None
 
 from . import utils
 from ._home import home