diff --git a/graphrag/general/extractor.py b/graphrag/general/extractor.py index 4c6f8e11f..c529380c5 100644 --- a/graphrag/general/extractor.py +++ b/graphrag/general/extractor.py @@ -193,7 +193,7 @@ class Extractor: if len(nodes) <= 1: return change.added_updated_nodes.add(nodes[0]) - change.removed_nodes.extend(nodes[1:]) + change.removed_nodes.update(nodes[1:]) nodes_set = set(nodes) node0_attrs = graph.nodes[nodes[0]] node0_neighbors = set(graph.neighbors(nodes[0]))