fix the fix

This commit is contained in:
svlandeg 2020-06-18 19:15:32 +02:00
parent e822367cf7
commit d1d6f16776

View File

@ -117,7 +117,7 @@ def json_to_annotations(doc):
if "lemma" in token:
lemmas.append(token["lemma"])
if "head" in token:
heads.append(token["head"])
heads.append(token["head"] + sent_start_i + i)
if "dep" in token:
labels.append(token["dep"])
# Ensure ROOT label is case-insensitive