Adds preserve_all_attrs to graph conversion call.

rapidsai · Jan 24, 2024 · 836d51f · 836d51f
1 parent 180baa8
commit 836d51f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/benchmarks/nx-cugraph/pytest-based/bench_algos.py b/benchmarks/nx-cugraph/pytest-based/bench_algos.py
@@ -233,7 +233,7 @@ def get_graph_obj_for_benchmark(graph_obj, backend_wrapper):
     """
     G = graph_obj
     if backend_wrapper.backend_name == "cugraph-preconverted":
-        G = nxcg.from_networkx(G)
+        G = nxcg.from_networkx(G, preserve_all_attrs=True)
     return G
 
 

diff --git a/benchmarks/nx-cugraph/pytest-based/run-2402.sh b/benchmarks/nx-cugraph/pytest-based/run-2402.sh
@@ -8,6 +8,7 @@
 
 mkdir -p logs
 
+# benches="$benches ..." pattern is easy to comment out individual runs
 benches=
 
 while [[ $1 != "" ]]; do