remove

generic_ner.py

@@ -476,6 +476,13 @@ def remove_included_entities(entities):
                     if entity["surface"] in other_entity["surface"]:
                         is_included = True
                         break
+                elif (
+                    entity["type"].split(".")[0] in other_entity["type"].split(".")[0]
+                    or other_entity["type"].split(".")[0]
+                    in entity["type"].split(".")[0]
+                ):
+                    if entity["surface"] in other_entity["surface"]:
+                        is_included = True
         if not is_included:
             final_entities.append(entity)
     return final_entities