changed all_pairs def

networkx · Feb 7, 2024 · 5bc797e · 5bc797e
1 parent e9c6c06
commit 5bc797e
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/nx_parallel/algorithms/approximation/connectivity.py b/nx_parallel/algorithms/approximation/connectivity.py
@@ -33,11 +33,7 @@ def _calculate_all_pairs_node_connectivity_subset(chunk):
 
     pairs = list(iter_func(nbunch, 2))
 
-    all_pairs = {}
-    for u, v in pairs:
-        all_pairs.setdefault(u, {})[v] = None
-        if not directed:
-            all_pairs.setdefault(v, {})[u] = None
+    all_pairs = {u: {v: None for v in nbunch if v != u} for u in nbunch}
 
     total_cores = nxp.cpu_count()
     num_in_chunk = max(len(pairs) // total_cores, 1)