From e6ce636e02ec1cd3f9893af6ab1eec4f113025db Mon Sep 17 00:00:00 2001
From: Stas Bekman <stas00@users.noreply.github.com>
Date: Sun, 7 Mar 2021 22:09:58 -0800
Subject: [PATCH] fix nltk lookup (#10585)

---
 examples/seq2seq/run_seq2seq.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/seq2seq/run_seq2seq.py b/examples/seq2seq/run_seq2seq.py
index db278073627..7450b652c28 100755
--- 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"