Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "5b9eacdf1bd093243cbe4d473825dcdd8682ffff"
Unverified Commit 61ffdeba authored by Nicolas Patry's avatar Nicolas Patry Committed by GitHub
Browse files

Fix ner average grouping with no groups (#24319) 

Fixes #https://github.com/huggingface/transformers/issues/24314
parent 34037129
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
src/transformers/pipelines/token_classification.py
View file @ 61ffdeba
...@@ -491,7 +491,8 @@ class TokenClassificationPipeline(ChunkPipeline): ...@@ -491,7 +491,8 @@ class TokenClassificationPipeline(ChunkPipeline):
word_entities.append(self.aggregate_word(word_group, aggregation_strategy)) word_entities.append(self.aggregate_word(word_group, aggregation_strategy))
word_group = [entity] word_group = [entity]
# Last item # 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 return word_entities
def group_sub_entities(self, entities: List[dict]) -> dict: def group_sub_entities(self, entities: List[dict]) -> dict:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment