From 9d7409ce71d19cb7f4a9f64c6cf75dc528c82053 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Tue, 3 Apr 2018 02:32:31 +0200 Subject: [PATCH] Hack at Alignment.flatten for split tokens --- spacy/_align.pyx | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/spacy/_align.pyx b/spacy/_align.pyx index cba117f94..1005b77b8 100644 --- a/spacy/_align.pyx +++ b/spacy/_align.pyx @@ -227,7 +227,12 @@ class Alignment(object): for head_val in head_vals: if not isinstance(head_val, tuple): head_val = (head_val, 0) - new.append(alignment[head_val]) + elif head_val not in alignment: + head_val = (head_val[0], 0) + try: + new.append(alignment[head_val]) + except: + raise return new