fix: failing unit tests, redundant / incomplete joblib param propogat…

…ion from ParallelConfig
networkx · Oct 17, 2024 · 427e5e7 · 427e5e7
1 parent 66a680f
commit 427e5e7
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/_nx_parallel/__init__.py b/_nx_parallel/__init__.py
@@ -84,7 +84,7 @@ def get_info():
                 },
             },
             "betweenness_centrality": {
-                "url": "https://github.com/networkx/nx-parallel/blob/main/nx_parallel/algorithms/centrality/betweenness.py#L20",
+                "url": "https://github.com/networkx/nx-parallel/blob/main/nx_parallel/algorithms/centrality/betweenness.py#L19",
                 "additional_docs": "The parallel computation is implemented by dividing the nodes into chunks and computing betweenness centrality for each chunk concurrently.",
                 "additional_parameters": {
                     'get_chunks : str, function (default = "chunks")': "A function that takes in a list of all the nodes as input and returns an iterable `node_chunks`. The default chunking is done by slicing the `nodes` into `n_jobs` number of chunks."
@@ -98,7 +98,7 @@ def get_info():
                 },
             },
             "edge_betweenness_centrality": {
-                "url": "https://github.com/networkx/nx-parallel/blob/main/nx_parallel/algorithms/centrality/betweenness.py#L96",
+                "url": "https://github.com/networkx/nx-parallel/blob/main/nx_parallel/algorithms/centrality/betweenness.py#L100",
                 "additional_docs": "The parallel computation is implemented by dividing the nodes into chunks and computing edge betweenness centrality for each chunk concurrently.",
                 "additional_parameters": {
                     'get_chunks : str, function (default = "chunks")': "A function that takes in a list of all the nodes as input and returns an iterable `node_chunks`. The default chunking is done by slicing the `nodes` into `n_jobs` number of chunks."

diff --git a/nx_parallel/interface.py b/nx_parallel/interface.py
@@ -1,8 +1,6 @@
-from functools import wraps
 from operator import attrgetter
 import networkx as nx
 from nx_parallel import algorithms
-from nx_parallel.utils.chunk import execute_parallel
 
 __all__ = ["BackendInterface", "ParallelGraph"]
 

diff --git a/nx_parallel/tests/test_get_chunks.py b/nx_parallel/tests/test_get_chunks.py
@@ -84,8 +84,9 @@ def iterator_func(G):
                     H, process_func, iterator_func, get_chunks=random_chunking
                 )
                 if isinstance(c1, types.GeneratorType):
-                    c1, c2 = list(c1), list(
-                        c2
+                    c1, c2 = (
+                        list(c1),
+                        list(c2),
                     )  # Convert generators to lists for comparison
                     if func in chk_dict_vals:
                         for i in range(len(G.nodes)):