diff --git a/src/input_parsers/wiktor_ner.py b/src/input_parsers/wiktor_ner.py index affbba4d8396e82523a2479a47f13906e4a8e302..8642bfa4043b8fc2290c7c973b2928eb795a3c8a 100644 --- a/src/input_parsers/wiktor_ner.py +++ b/src/input_parsers/wiktor_ner.py @@ -79,7 +79,7 @@ class WiktorNERInputParser(InputParser): annotations = [] # Morphosyntactic annotations - if content.tokens: + if content.tokens(): for token in content.tokens(): if token.lexemes: for lexeme in token.lexemes: @@ -99,7 +99,7 @@ class WiktorNERInputParser(InputParser): ) ) # NER annotations - if content.spans: + if 'ner' in content.get_span_types(): for entity in content.spans('ner'): if entity.type: annotations.append(