Commit 7f983e2
authored
Fix spacy test. (Kaggle#1008)
spacy has been upgraded to 3.x (fastai had a 2.x requirement but now
support 3.x which upgraded spacy.)1 parent 8d00db0 commit 7f983e2
2 files changed
Lines changed: 2 additions & 2 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -350,7 +350,7 @@ RUN pip install bleach && \ | |||
| 350 | 350 | pip install allennlp && \ | |
| 351 | 351 | # https://b.corp.google.com/issues/184685619#comment9: 3.9.0 is causing a major performance degradation with spacy 2.3.5 | |
| 352 | 352 | pip install importlib-metadata==3.4.0 && \ | |
| 353 | - python -m spacy download en && python -m spacy download en_core_web_lg && \ | ||
| 353 | + python -m spacy download en_core_web_sm && python -m spacy download en_core_web_lg && \ | ||
| 354 | 354 | apt-get install -y ffmpeg && \ | |
| 355 | 355 | /tmp/clean-layer.sh | |
| 356 | 356 | ||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -4,6 +4,6 @@ | |||
| 4 | 4 | ||
| 5 | 5 | class TestSpacy(unittest.TestCase): | |
| 6 | 6 | def test_model(self): | |
| 7 | - nlp = spacy.load('en') | ||
| 7 | + nlp = spacy.load('en_core_web_sm') | ||
| 8 | 8 | doc = nlp('This is a sentence.') | |
| 9 | 9 | self.assertEqual(5, len(doc)) | |
0 commit comments