SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Catch import datasets common errors (#7456)

Browse Source
This commit is contained in:
Sylvain Gugger
2020-09-29 13:42:09 -04:00
committed by GitHub
parent f1220c5fe2
commit d0fd7154c5
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/transformers/file_utils.py
View File

@@ -68,8 +68,12 @@ except (ImportError, AssertionError):
try: try:
import datasets # noqa: F401 import datasets # noqa: F401
_datasets_available = True # Check we're not importing a "datasets" directory somewhere
_datasets_available = hasattr(datasets, "__version__") and hasattr(datasets, "load_dataset")
if _datasets_available:
logger.debug(f"Succesfully imported datasets version {datasets.__version__}") logger.debug(f"Succesfully imported datasets version {datasets.__version__}")
else:
logger.debug("Imported a datasets object but this doesn't seem to be the 🤗 datasets library.")
except ImportError: except ImportError:
_datasets_available = False _datasets_available = False