Skip to content

Commit

Permalink
Fix MG Betweenness Centrality Test Bug (#4429)
Browse files Browse the repository at this point in the history
Fixes bug seen in MG tests due to incorrect call to `enable_batch`

Authors:
  - Ralph Liu (https://github.com/nv-rliu)

Approvers:
  - Alex Barghi (https://github.com/alexbarghi-nv)

URL: #4429
  • Loading branch information
nv-rliu authored May 20, 2024
1 parent 78b380e commit f1490b0
Showing 1 changed file with 5 additions and 10 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -111,18 +111,13 @@ def calc_betweenness_centrality(
else:
edge_attr = None

G = None
G = graph_file.get_graph(
download=True,
create_using=cugraph.Graph(directed=directed),
ignore_weights=not edgevals,
)
if multi_gpu_batch:
G = graph_file.get_dask_graph(
create_using=cugraph.Graph(directed=directed), ignore_weights=not edgevals
)
G.enable_batch()
else:
G = graph_file.get_graph(
download=True,
create_using=cugraph.Graph(directed=directed),
ignore_weights=not edgevals,
)

M = G.to_pandas_edgelist().rename(
columns={"src": "0", "dst": "1", "wgt": edge_attr}
Expand Down

0 comments on commit f1490b0

Please sign in to comment.