Fix the entity creation in global_matchers

eriknovak · Jul 23, 2024 · 37efe07 · 37efe07
1 parent 4b11866
commit 37efe07
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/anonipy/anonymize/extractors/pattern_extractor.py b/anonipy/anonymize/extractors/pattern_extractor.py
@@ -192,6 +192,8 @@ def global_matchers(doc: Doc) -> None:
                     # define the entity span
                     start, end = match.span(1)
                     entity = doc.char_span(start, end, label=label["label"])
+                    if not entity:
+                        continue
                     entity._.score = 1.0
                     # add the entity to the previous entity list
                     prev_entities = self._get_doc_entity_spans(doc)