diff --git a/nxbench/benchmarks/benchmark.py b/nxbench/benchmarks/benchmark.py index c80ce30..cb2a74a 100644 --- a/nxbench/benchmarks/benchmark.py +++ b/nxbench/benchmarks/benchmark.py @@ -199,13 +199,14 @@ def prepare_benchmark( if "graphblas" in backend and is_graphblas_available(): try: + gb = import_module("graphblas") ga = import_module("graphblas_algorithms") except ImportError: logger.exception("graphblas_algorithms backend not available") return None try: logger.info( - f"GraphBlas Algorithms nthreads={ga.ss.config['nthreads']} " + f"GraphBlas Algorithms nthreads={gb.ss.config['nthreads']} " ) return ga.Graph.from_networkx(original_graph) except Exception: diff --git a/nxbench/configs/example.yaml b/nxbench/configs/example.yaml index 76fe855..63517d0 100644 --- a/nxbench/configs/example.yaml +++ b/nxbench/configs/example.yaml @@ -241,6 +241,7 @@ validation: matrix: backend: - "networkx" + - "parallel" - "graphblas" num_threads: - "1" @@ -250,6 +251,7 @@ matrix: env_config: req: - "networkx==3.4.2" + - "nx-parallel-0.3rc0.dev0" - "graphblas_algorithms==2023.10.0" pythons: - "3.10"