Fix lazy init to stop hiding errors in import (#14124)

This commit is contained in:
Sylvain Gugger 2021-10-25 16:53:47 -04:00 committed by GitHub
parent c99a2832ed
commit 8560b55b5e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -2146,7 +2146,12 @@ class _LazyModule(ModuleType):
return value
def _get_module(self, module_name: str):
return importlib.import_module("." + module_name, self.__name__)
try:
return importlib.import_module("." + module_name, self.__name__)
except Exception as e:
raise RuntimeError(
f"Failed to import {self.__name__}.{module_name} because of the following error (look up to see its traceback):\n{e}"
) from e
def __reduce__(self):
return (self.__class__, (self._name, self.__file__, self._import_structure))