explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-10-31 07:57:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix error catching

Browse Source
This commit is contained in:
svlandeg 2020-06-15 12:56:32 +02:00
parent b9c9cbb2cd
commit b4d914ec77
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
spacy/tokens/doc.pyx
View File

@ -699,12 +699,12 @@ cdef class Doc:
# Handle inputs like doc.to_array(ORTH)
py_attr_ids = [py_attr_ids]
# Allow strings, e.g. 'lemma' or 'LEMMA'
py_attr_ids = [id_.upper() if hasattr(id_, "upper") else id_ for id_ in py_attr_ids]
for key in py_attr_ids:
if key not in IDS:
try:
py_attr_ids = [(IDS[id_.upper()] if hasattr(id_, "upper") else id_)
for id_ in py_attr_ids]
except KeyError as msg:
keys = [k for k in IDS.keys() if not k.startswith("FLAG")]
raise ValueError(Errors.E983.format(dict_name="IDS", key=key, keys=keys))
py_attr_ids = [IDS[id_] for id_ in py_attr_ids]
raise KeyError(Errors.E983.format(dict_name="IDS", key=msg, keys=keys))
# Make an array from the attributes --- otherwise our inner loop is
# Python dict iteration.
cdef np.ndarray attr_ids = numpy.asarray(py_attr_ids, dtype="i")