Another method of removing entities.

dan-zeman · dan-zeman · commit 16c3a48ed3eb · 2023-02-17T22:27:19.000+01:00
diff --git a/udapi/block/corefud/removenocorefentities.py b/udapi/block/corefud/removenocorefentities.py
@@ -18,4 +18,4 @@ def process_document(self, doc):
         entities = doc.coref_entities
         if not entities:
             return
-        doc.coref_entities = [e for e in entities if not re.match(r'^NOCOREF', e.eid)]
+        doc._eid_to_entity = {e._eid: e for e in entities if not re.match(r'^NOCOREF', e.eid)}
