diff --git a/ci/wheel_smoke_test_nx-cugraph.py b/ci/wheel_smoke_test_nx-cugraph.py index eec1ccd4d9a..4f924001b82 100644 --- a/ci/wheel_smoke_test_nx-cugraph.py +++ b/ci/wheel_smoke_test_nx-cugraph.py @@ -30,11 +30,9 @@ # supports the "backend" kwarg may not be available in the testing env. nxcu_result = nx_cugraph.betweenness_centrality(G) - nx_nodes = nx_result.keys() - nxcu_nodes = nxcu_result.keys() + nx_nodes, nxcu_nodes = nx_result.keys(), nxcu_result.keys() assert nxcu_nodes == nx_nodes for node_id in nx_nodes: - nx_bc = nx_result[node_id] - nxcu_bc = nxcu_result[node_id] - assert math.isclose(nx_bc, nxcu_bc, rel_tol=1e-7), \ - f"bc for {node_id=} are not close enough: {nx_bc=}, {nxcu_bc=}" + nx_bc, nxcu_bc = nx_result[node_id], nxcu_result[node_id] + assert math.isclose(nx_bc, nxcu_bc, rel_tol=1e-6), \ + f"bc for {node_id=} exceeds tolerance: {nx_bc=}, {nxcu_bc=}"