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

Exclude Databricks from notebook env (#17496)

Browse Source
This commit is contained in:
Sylvain Gugger
2022-06-01 09:00:11 -04:00
committed by GitHub
parent 3042ea4f6f
commit 6813439fdc
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/utils/import_utils.py
View File

@@ -460,6 +460,8 @@ def is_in_notebook():
raise ImportError("console") raise ImportError("console")
if "VSCODE_PID" in os.environ: if "VSCODE_PID" in os.environ:
raise ImportError("vscode") raise ImportError("vscode")
if "DATABRICKS_RUNTIME_VERSION" in os.environ:
raise ImportError("databricks")
return importlib.util.find_spec("IPython") is not None return importlib.util.find_spec("IPython") is not None
except (AttributeError, ImportError, KeyError): except (AttributeError, ImportError, KeyError):