fix issue with ntlk

c53fdacb · Andres Marafioti · ba40056a · c53fdacb
Commit c53fdacb authored 7 months ago by Andres Marafioti
--- a/s2s_pipeline.py
+++ b/s2s_pipeline.py
@@ -40,7 +40,7 @@ from utils import (
 # Ensure that the necessary NLTK resources are available
 try:
    nltk.data.find('tokenizers/punkt_tab')
-except LookupError:
+except (LookupError, OSError):
    nltk.download('punkt_tab')

 # caching allows ~50% compilation time reduction