Skip to content

Commit

Permalink
Adds comments to smoke test script.
Browse files Browse the repository at this point in the history
  • Loading branch information
rlratzel committed Sep 8, 2023
1 parent e473c5b commit e67a3ee
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions ci/wheel_smoke_test_nx-cugraph.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,14 @@
if __name__ == "__main__":
G = nx.Graph()
G.add_edges_from([(0, 1), (1, 2), (2, 3)])

nx_result = nx.betweenness_centrality(G)
# nx_cugraph is intended to be called via the NetworkX dispatcher, like
# this:
# nxcu_result = nx.betweenness_centrality(G, backend="cugraph")
#
# but here it is being called directly since the NetworkX version that
# supports the "backend" kwarg may not be available in the testing env.
nxcu_result = nx_cugraph.betweenness_centrality(G)

nx_nodes = nx_result.keys()
Expand Down

0 comments on commit e67a3ee

Please sign in to comment.