feat: additional teardown handling

dPys · Dec 16, 2024 · e76e761 · e76e761
1 parent 8109dd9
commit e76e761
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nxbench/benchmarks/benchmark.py b/nxbench/benchmarks/benchmark.py
@@ -181,6 +181,8 @@ def prepare_benchmark(
             except ImportError:
                 logger.exception("nx-parallel backend not available")
                 return None
+            nx.config.backends.parallel.active = True
+            nx.config.backends.parallel.n_jobs = num_thread
             return nxp.ParallelGraph(original_graph)
 
         if "cugraph" in backend and is_nx_cugraph_available():