Update nx_cugraph/algorithms/bipartite/centrality.py

Co-authored-by: Erik Welch <[email protected]>
rapidsai · Nov 19, 2024 · 090680f · 090680f
1 parent 0c0b90d
commit 090680f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nx_cugraph/algorithms/bipartite/centrality.py b/nx_cugraph/algorithms/bipartite/centrality.py
@@ -38,7 +38,7 @@ def betweenness_centrality(G, nodes):
         do_expensive_check=False,
     )
     top_node_ids = G._nodekeys_to_nodearray(set(nodes))
-    bottom_node_ids = cp.setdiff1d(
+    bottom_node_ids = cp.delete(
         cp.arange(G._N, dtype=top_node_ids.dtype), top_node_ids
     )
     n = top_node_ids.size