Improve error message when extension is not found (#795)

* Improve error message when extension is not found
pytorch · May 27, 2020 · 3fbe4da · 3fbe4da
1 parent 81d32c1
commit 3fbe4da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchtext/__init__.py b/torchtext/__init__.py
@@ -32,7 +32,7 @@ def _init_extension():
  extfinder = importlib.machinery.FileFinder(lib_dir, loader_details)
  ext_specs = extfinder.find_spec("_torchtext")
  if ext_specs is None:
- raise ImportError
+ raise ImportError("torchtext C++ Extension is not found.")
  torch.ops.load_library(ext_specs.origin)
  torch.classes.load_library(ext_specs.origin)