new: run_on_gpu dev param

arangodb · Apr 28, 2024 · 4797eb9 · 4797eb9
1 parent 498d115
commit 4797eb9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nx_arangodb/algorithms/centrality/betweenness.py b/nx_arangodb/algorithms/centrality/betweenness.py
@@ -27,12 +27,12 @@
     _plc="betweenness_centrality",
 )
 def betweenness_centrality(
-    G, k=None, normalized=True, weight=None, endpoints=False, seed=None
+    G, k=None, normalized=True, weight=None, endpoints=False, seed=None, run_on_gpu=True
 ):
     print("ANTHONY: Calling betweenness_centrality from nx_arangodb")
 
     # 1.
-    if GPU_ENABLED:
+    if GPU_ENABLED and run_on_gpu:
         print("ANTHONY: GPU is enabled. Using nx-cugraph bc()")
 
         if weight is not None: