Take out last entity handling to avoid redundancy with recently merge…

…d PR (huggingface#5439)
enzoampil · Jul 4, 2020 · af54559 · af54559
1 parent 4a7a483
commit af54559
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/transformers/pipelines.py b/src/transformers/pipelines.py
@@ -1044,9 +1044,6 @@ def __call__(self, *args, **kwargs):
                     else:
                         entity_groups += [self.group_entities(entity_group_disagg)]
                         entity_group_disagg = [entity]
-                        # If it's the last entity,
-                        if is_last_idx:
-                            entity_groups += [self.group_entities(entity_group_disagg)]
 
                 entities += [entity]