Fix spacy test. (#1008)

spacy has been upgraded to 3.x (fastai had a 2.x requirement but now support 3.x which upgraded spacy.)
Kaggle · May 6, 2021 · 7f983e2 · 7f983e2
1 parent 8d00db0
commit 7f983e2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -350,7 +350,7 @@ RUN pip install bleach && \
  pip install allennlp && \
  # https://b.corp.google.com/issues/184685619#comment9: 3.9.0 is causing a major performance degradation with spacy 2.3.5
  pip install importlib-metadata==3.4.0 && \
- python -m spacy download en && python -m spacy download en_core_web_lg && \
+ python -m spacy download en_core_web_sm && python -m spacy download en_core_web_lg && \
  apt-get install -y ffmpeg && \
  /tmp/clean-layer.sh
 

diff --git a/tests/test_spacy.py b/tests/test_spacy.py
@@ -4,6 +4,6 @@
 
 class TestSpacy(unittest.TestCase):
  def test_model(self):
- nlp = spacy.load('en')
+ nlp = spacy.load('en_core_web_sm')
  doc = nlp('This is a sentence.')
  self.assertEqual(5, len(doc))