From de14d8d2aecb774d1f5e534ab56171547029ccb7 Mon Sep 17 00:00:00 2001 From: Rick Ratzel Date: Fri, 8 Sep 2023 14:30:58 -0500 Subject: [PATCH] Cleanup. --- ci/wheel_smoke_test_nx-cugraph.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) 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=}"