"...research_projects/quantization-qdqbert/run_quant_qa.py" did not exist on "1d30ec95c7a3656972a394d28a86efdb6ca4cdab"
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
...@@ -491,6 +491,7 @@ class TokenClassificationPipeline(ChunkPipeline): ...@@ -491,6 +491,7 @@ 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
if word_group is not None:
word_entities.append(self.aggregate_word(word_group, aggregation_strategy)) word_entities.append(self.aggregate_word(word_group, aggregation_strategy))
return word_entities return word_entities
......
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