fix nltk lookup (#10585)

e6ce636e · Stas Bekman · GitHub · 9dd054fb · e6ce636e
Unverified Commit e6ce636e authored Mar 07, 2021 by Stas Bekman Committed by GitHub Mar 07, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/seq2seq/run_seq2seq.py examples/seq2seq/run_seq2seq.py +1 -1

No files found.
--- a/examples/seq2seq/run_seq2seq.py
+++ b/examples/seq2seq/run_seq2seq.py
@@ -52,7 +52,7 @@ logger = logging.getLogger(__name__)

 try:
    nltk.data.find("tokenizers/punkt")
-except LookupError:
+except (LookupError, OSError):
    if is_offline_mode():
        raise LookupError(
            "Offline mode: run this script without TRANSFORMERS_OFFLINE first to download nltk data files"