mirror of
https://github.com/explosion/spaCy.git
synced 2025-04-27 12:23:42 +03:00
Fix beam parse model test
This commit is contained in:
parent
b38a8f4a94
commit
0adadcb3f0
|
@ -1,10 +1,11 @@
|
||||||
import spacy
|
# coding: utf8
|
||||||
|
from __future__ import unicode_literals
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
@pytest.mark.models
|
|
||||||
def test_beam_parse():
|
|
||||||
nlp = spacy.load('en_core_web_sm')
|
|
||||||
doc = nlp(u'Australia is a country', disable=['ner'])
|
|
||||||
ents = nlp.entity(doc, beam_width=2)
|
|
||||||
print(ents)
|
|
||||||
|
|
||||||
|
@pytest.mark.models('en')
|
||||||
|
def test_beam_parse(EN):
|
||||||
|
doc = EN(u'Australia is a country', disable=['ner'])
|
||||||
|
ents = EN.entity(doc, beam_width=2)
|
||||||
|
print(ents)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user