From d0fd7154c599cd282e6dc7e060b87b90f1e06c93 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Tue, 29 Sep 2020 13:42:09 -0400 Subject: [PATCH] Catch import datasets common errors (#7456) --- src/transformers/file_utils.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/transformers/file_utils.py b/src/transformers/file_utils.py index 9c9578dd48..f3844cef79 100644 --- a/src/transformers/file_utils.py +++ b/src/transformers/file_utils.py @@ -68,8 +68,12 @@ except (ImportError, AssertionError): try: import datasets # noqa: F401 - _datasets_available = True - logger.debug(f"Succesfully imported datasets version {datasets.__version__}") + # 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__}") + else: + logger.debug("Imported a datasets object but this doesn't seem to be the 🤗 datasets library.") except ImportError: _datasets_available = False