Use sets instead of dictionaries in merge_nodes

2026-07-15 11:25:53 +08:00 · 2014-06-28 21:02:25 +05:30
parent dd67b3fce7
commit 6abf194dd6
1 changed files with 2 additions and 11 deletions
@@ -63,17 +63,8 @@ class RAG(nx.Graph):
            The dict of keyword arguments passed to the `weight_func`.

        """
-        neighbors = self.adj[src].copy()
-        neighbors.update(self.adj[dst])
-
-        try:
-            del neighbors[src]
-        except KeyError:
-            pass
-        try:
-            del neighbors[dst]
-        except KeyError:
-            pass
+        neighbors = (set(self.neighbors(src)) & set(
+            self.neighbors(dst))) - set([src, dst])

        for neighbor in neighbors:
            w = weight_func(self, src, dst, neighbor, *extra_arguments,