explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-25 09:26:27 +03:00
Code Issues Packages Projects Releases Wiki Activity

* Fix import of attrs in doc.pyx, and update the get_token_attr function.

Browse Source
This commit is contained in:
Matthew Honnibal 2015-07-16 01:15:34 +02:00
parent 65251e7625
commit 9d956b07e9
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
spacy/tokens/doc.pyx
View File

@ -6,10 +6,9 @@ import numpy
from ..lexeme cimport EMPTY_LEXEME from ..lexeme cimport EMPTY_LEXEME
from ..serialize import BitArray from ..serialize import BitArray
from ..strings cimport slice_unicode from ..strings cimport slice_unicode
from ..typedefs cimport attr_id_t, attr_t, flags_t from ..attrs cimport attr_id_t, attr_t, flags_t
from ..typedefs cimport LEMMA from ..attrs cimport ID, ORTH, NORM, LOWER, SHAPE, PREFIX, SUFFIX, LENGTH, CLUSTER
from ..typedefs cimport ID, ORTH, NORM, LOWER, SHAPE, PREFIX, SUFFIX, LENGTH, CLUSTER from ..attrs cimport POS, LEMMA, TAG, DEP, HEAD, SPACY, ENT_IOB, ENT_TYPE
from ..typedefs cimport POS, LEMMA, TAG, DEP
from ..parts_of_speech import UNIV_POS_NAMES from ..parts_of_speech import UNIV_POS_NAMES
from ..parts_of_speech cimport CONJ, PUNCT from ..parts_of_speech cimport CONJ, PUNCT
from ..lexeme cimport check_flag from ..lexeme cimport check_flag
@ -37,6 +36,14 @@ cdef attr_t get_token_attr(const TokenC* token, attr_id_t feat_name) nogil:
return token.tag return token.tag
elif feat_name == DEP: elif feat_name == DEP:
return token.dep return token.dep
elif feat_name == HEAD:
return token.head
elif feat_name == SPACY:
return token.spacy
elif feat_name == ENT_IOB:
return token.ent_iob
elif feat_name == ENT_TYPE:
return token.ent_type
else: else:
return get_lex_attr(token.lex, feat_name) return get_lex_attr(token.lex, feat_name)