Fix add_attrs for cluster

2024-09-21 03:19:13 +03:00 · 2017-10-30 16:07:50 +01:00 · 2017-10-30 16:07:50 +01:00 · 41d0f1665a
commit 41d0f1665a
parent 5ede7cec9b
1 changed files with 2 additions and 0 deletions
--- a/spacy/lexeme.pyx
+++ b/spacy/lexeme.pyx
@ -76,6 +76,8 @@ cdef class Lexeme:
        for attr, value in attrs.items():
            if attr == PROB:
                self.c.prob = value
+            elif attr == CLUSTER:
+                self.c.cluster = int(value)
            elif isinstance(value, int) or isinstance(value, long):
                Lexeme.set_struct_attr(self.c, attr, value)
            else: