Skip to content

GitLab

Unverified Commit 21c88a07 authored by Thomas Wolf's avatar Thomas Wolf Committed by GitHub
Browse files

Merge pull request #341 from potatochip/patch-1 

catch exception if pathlib not install
parents 3c01dfb7 0c970caa
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pytorch_pretrained_bert/file_utils.py
View file @ 21c88a07
...@@ -29,7 +29,7 @@ try: ...@@ -29,7 +29,7 @@ try:
from pathlib import Path from pathlib import Path
PYTORCH_PRETRAINED_BERT_CACHE = Path(os.getenv('PYTORCH_PRETRAINED_BERT_CACHE', PYTORCH_PRETRAINED_BERT_CACHE = Path(os.getenv('PYTORCH_PRETRAINED_BERT_CACHE',
Path.home() / '.pytorch_pretrained_bert')) Path.home() / '.pytorch_pretrained_bert'))
except AttributeError: except (AttributeError, ImportError):
PYTORCH_PRETRAINED_BERT_CACHE = os.getenv('PYTORCH_PRETRAINED_BERT_CACHE', PYTORCH_PRETRAINED_BERT_CACHE = os.getenv('PYTORCH_PRETRAINED_BERT_CACHE',
os.path.join(os.path.expanduser("~"), '.pytorch_pretrained_bert')) os.path.join(os.path.expanduser("~"), '.pytorch_pretrained_bert'))
......
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