fix: add missing vanilla graphblas import

dPys · Dec 11, 2024 · a3731a9 · a3731a9
1 parent 0e95e67
commit a3731a9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git 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
@@ -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"