catch exception if pathlib not install

0c970caa · Aaron Mangum · GitHub · 2152bfea · 0c970caa
Unverified Commit 0c970caa authored Mar 04, 2019 by Aaron Mangum Committed by GitHub Mar 04, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

pytorch_pretrained_bert/file_utils.py pytorch_pretrained_bert/file_utils.py +1 -1

No files found.
--- a/pytorch_pretrained_bert/file_utils.py
+++ b/pytorch_pretrained_bert/file_utils.py
@@ -29,7 +29,7 @@ try:
    from pathlib import Path
    PYTORCH_PRETRAINED_BERT_CACHE = Path(os.getenv('PYTORCH_PRETRAINED_BERT_CACHE',
                                                   Path.home() / '.pytorch_pretrained_bert'))
-except AttributeError:
+except (AttributeError, ImportError):
    PYTORCH_PRETRAINED_BERT_CACHE = os.getenv('PYTORCH_PRETRAINED_BERT_CACHE',
                                              os.path.join(os.path.expanduser("~"), '.pytorch_pretrained_bert'))