Update graphrag/entity_resolution.py

infiniflow · Sep 30, 2024 · e63750b · e63750b
1 parent abaa877
commit e63750b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/graphrag/entity_resolution.py b/graphrag/entity_resolution.py
@@ -98,7 +98,7 @@ def __call__(self, graph: nx.Graph, prompt_variables: dict[str, Any] | None = No
             for a, b in itertools.permutations(v, 2):
                 if self.is_similarity(a, b) and (b, a) not in candidate_resolution_tmp:
                     candidate_resolution_tmp.append((a, b))
-            candidate_resolution[k] = candidate_resolution_tmp or v
+            candidate_resolution[k] = candidate_resolution_tmp
 
         gen_conf = {"temperature": 0.5}
         resolution_result = set()