darabos commited on
Commit
8a4f65f
·
1 Parent(s): d6c271c

Do not hardcode cuGraph backend.

Browse files
lynxkite-graph-analytics/src/lynxkite_graph_analytics/lynxkite_ops.py CHANGED
@@ -229,7 +229,7 @@ def compute_pagerank(graph: nx.Graph, *, damping=0.85, iterations=100):
229
  @op("Compute betweenness centrality")
230
  @nx_node_attribute_func("betweenness_centrality")
231
  def compute_betweenness_centrality(graph: nx.Graph, *, k=10):
232
- return nx.betweenness_centrality(graph, k=k, backend="cugraph")
233
 
234
 
235
  @op("Discard loop edges")
 
229
  @op("Compute betweenness centrality")
230
  @nx_node_attribute_func("betweenness_centrality")
231
  def compute_betweenness_centrality(graph: nx.Graph, *, k=10):
232
+ return nx.betweenness_centrality(graph, k=k)
233
 
234
 
235
  @op("Discard loop edges")