Small fixes to NotebookProgressCallback (#7813)

2020-10-15 10:30:34 -04:00
parent 6f45dd2fac
commit 2ce3ddab2d
2 changed files with 24 additions and 5 deletions
--- a/src/transformers/file_utils.py
+++ b/src/transformers/file_utils.py
@@ -153,7 +153,7 @@ try:
    import IPython  # noqa: F401

    _in_notebook = True
-except:  # noqa: E722
+except (AttributeError, ImportError, KeyError):
    _in_notebook = False