From e76e761794b6104ef07aac2a461f96e679ab9dfa Mon Sep 17 00:00:00 2001
From: dpys <dpysalexander@gmail.com>
Date: Mon, 16 Dec 2024 16:11:54 -0500
Subject: [PATCH] feat: additional teardown handling

---
 nxbench/benchmarks/benchmark.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/nxbench/benchmarks/benchmark.py b/nxbench/benchmarks/benchmark.py
index 370b50c..50aae0f 100644
--- 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():