removed entity label

neo4j-labs · Sep 16, 2024 · 8f0a706 · 8f0a706
1 parent 801745e
commit 8f0a706
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/backend/src/chunkid_entities.py b/backend/src/chunkid_entities.py
@@ -95,6 +95,10 @@ def remove_duplicate_nodes(nodes,property="element_id"):
     for node in nodes:
         element_id = node[property]
         if element_id not in seen_element_ids:
+            if "labels" in node.keys():
+                labels = set(node["labels"])
+                labels.discard("__Entity__")
+                node["labels"] = list(labels)
             unique_nodes.append(node)
             seen_element_ids.add(element_id)