checkpoint

pytorch · Feb 19, 2021 · 178b618 · 178b618
1 parent db0179c
commit 178b618
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/packaging/torchtext/meta.yaml b/packaging/torchtext/meta.yaml
@@ -41,6 +41,7 @@ test:
   requires:
     - pytest
     - cpuonly
+    - nltk
 
 about:
   home: https://github.com/pytorch/text

diff --git a/torchtext/datasets/sst.py b/torchtext/datasets/sst.py
@@ -5,10 +5,7 @@
 from torchtext.datasets.common import find_match
 import os
 import io
-try:
-    from nltk.tree import Tree
-except ImportError:
-    print("Please install NLTK. See the docs at https://nltk.org for more information.")
+from nltk.tree import Tree
 
 
 URL = 'http://nlp.stanford.edu/sentiment/trainDevTestTrees_PTB.zip'