dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix ner average grouping with no groups (#24319) 

Fixes #https://github.com/huggingface/transformers/issues/24314
This commit is contained in:
Nicolas Patry 2023-06-16 16:43:19 +02:00 committed by GitHub
parent 3403712958
commit 61ffdeba38
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/pipelines/token_classification.py
View File

@ -491,7 +491,8 @@ class TokenClassificationPipeline(ChunkPipeline):
word_entities.append(self.aggregate_word(word_group, aggregation_strategy))
word_group = [entity]
# Last item
word_entities.append(self.aggregate_word(word_group, aggregation_strategy))
if word_group is not None:
word_entities.append(self.aggregate_word(word_group, aggregation_strategy))
return word_entities
def group_sub_entities(self, entities: List[dict]) -> dict: