Fix ner average grouping with no groups (#24319)

Fixes #https://github.com/huggingface/transformers/issues/24314

Fix ner average grouping with no groups (#24319)
Fixes #https://github.com/huggingface/transformers/issues/24314
61ffdeba · Nicolas Patry · GitHub · 34037129 · 61ffdeba
Unverified Commit 61ffdeba authored Jun 16, 2023 by Nicolas Patry Committed by GitHub Jun 16, 2023
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/transformers/pipelines/token_classification.py src/transformers/pipelines/token_classification.py +2 -1

No files found.
--- a/src/transformers/pipelines/token_classification.py
+++ b/src/transformers/pipelines/token_classification.py
@@ -491,6 +491,7 @@ class TokenClassificationPipeline(ChunkPipeline):
                word_entities.append(self.aggregate_word(word_group, aggregation_strategy))
                word_group = [entity]
        # Last item
+        if word_group is not None:
            word_entities.append(self.aggregate_word(word_group, aggregation_strategy))
        return word_entities