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