Support TAG attribute in doc.from_array

2017-01-10 22:47:07 +01:00 · 2017-01-10 22:47:07 +01:00 · 44e2b0100d
parent 271997a146
commit 44e2b0100d
1 changed files with 3 additions and 0 deletions
--- a/spacy/tokens/doc.pyx
+++ b/spacy/tokens/doc.pyx
@ -585,6 +585,9 @@ cdef class Doc:
            elif attr_id == POS:
                for i in range(length):
                    tokens[i].pos = <univ_pos_t>values[i]
+            elif attr_id == TAG:
+                for i in range(length):
+                    tokens[i].tag = <univ_pos_t>values[i]
            elif attr_id == DEP:
                for i in range(length):
                    tokens[i].dep = values[i]