add comp back

generic_ner.py

@@ -513,9 +513,10 @@ def refine_entities_with_coarse(all_entities, coarse_entities):
                 entity["type"] = coarse_entity[
                     "type"
                 ]  # Update the type if the confidence is higher
-            else:
-                entity["type"] = entity["type"].split('.')[0]
+
     # No need to append to refined_entities, we're modifying in place
+    for entity in all_entities:
+        entity["type"] = entity["type"].split(".")[0]
     return all_entities