From 20b8ccd1e94c5b11b511cc53c1dd4099cee16925 Mon Sep 17 00:00:00 2001
From: Yue-Lyu123 <61424174+Yue-Lyu123@users.noreply.github.com>
Date: Tue, 1 Apr 2025 12:06:28 +0800
Subject: [PATCH] Hotfix ece5903 (#6705)

I'm really sorry, I found that in graphrag/general/extractor.py under
def __call__, the line change.removed_nodes.extend(nodes[1:]) causes an
AttributeError: 'set' object has no attribute 'extend'. Could you please
merge the branch e666528 again? I made some modifications.
---
 graphrag/general/extractor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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]))