Unverified Commit d0fd7154 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Catch import datasets common errors (#7456)

parent f1220c5f
...@@ -68,8 +68,12 @@ except (ImportError, AssertionError): ...@@ -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
logger.debug(f"Succesfully imported datasets version {datasets.__version__}") _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: except ImportError:
_datasets_available = False _datasets_available = False
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment