Fix ArcEager for missing values

2017-08-01 22:10:05 +02:00 · 2017-08-01 22:10:05 +02:00 · 8fce187de4
parent c16ef0a85c
commit 8fce187de4
1 changed files with 1 additions and 1 deletions
--- a/spacy/syntax/arc_eager.pyx
+++ b/spacy/syntax/arc_eager.pyx
@ -362,7 +362,7 @@ cdef class ArcEager(TransitionSystem):
        if not self.has_gold(gold):
            return None
        for i in range(gold.length):
-            if gold.heads[i] is None: # Missing values
+            if gold.heads[i] is None or gold.labels[i] is None: # Missing values
                gold.c.heads[i] = i
                gold.c.has_dep[i] = False
            else: