Commit b8a3a13bf2ad0c76b5d8179984c6fc4370d86848
1 parent
65a85924
Minor preparator fix.
Showing
1 changed file
with
3 additions
and
1 deletions
preparator.py
@@ -169,7 +169,9 @@ def get_sets(mentions): | @@ -169,7 +169,9 @@ def get_sets(mentions): | ||
169 | all_mentions = [] | 169 | all_mentions = [] |
170 | clustered_mensions = [] | 170 | clustered_mensions = [] |
171 | for mention in mentions: | 171 | for mention in mentions: |
172 | - all_mentions.append(mention.attrib['span']) | 172 | + if not mention.attrib['mention_head'] in POSSIBLE_HEADS: |
173 | + all_mentions.append(mention.attrib['span']) | ||
174 | + | ||
173 | set_id = mention.attrib['mention_group'] | 175 | set_id = mention.attrib['mention_group'] |
174 | if set_id == 'empty' or set_id == '' or mention.attrib['mention_head'] in POSSIBLE_HEADS: | 176 | if set_id == 'empty' or set_id == '' or mention.attrib['mention_head'] in POSSIBLE_HEADS: |
175 | pass | 177 | pass |