From b8a3a13bf2ad0c76b5d8179984c6fc4370d86848 Mon Sep 17 00:00:00 2001 From: Bartlomiej Niton <bartek.niton@gmail.com> Date: Mon, 21 Aug 2017 18:47:42 +0200 Subject: [PATCH] Minor preparator fix. --- preparator.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/preparator.py b/preparator.py index f9ac0c7..a705a5c 100644 --- a/preparator.py +++ b/preparator.py @@ -169,7 +169,9 @@ def get_sets(mentions): all_mentions = [] clustered_mensions = [] for mention in mentions: - all_mentions.append(mention.attrib['span']) + if not mention.attrib['mention_head'] in POSSIBLE_HEADS: + all_mentions.append(mention.attrib['span']) + set_id = mention.attrib['mention_group'] if set_id == 'empty' or set_id == '' or mention.attrib['mention_head'] in POSSIBLE_HEADS: pass -- libgit2 0.22.2