From 247d8d2ff482c5fd32bebf195f01796e4637834d Mon Sep 17 00:00:00 2001 From: Alexandria Barghi Date: Fri, 22 Sep 2023 11:13:59 -0700 Subject: [PATCH] revert dataloader change --- python/cugraph-dgl/tests/test_dataloader.py | 1 - 1 file changed, 1 deletion(-) diff --git a/python/cugraph-dgl/tests/test_dataloader.py b/python/cugraph-dgl/tests/test_dataloader.py index b9ff0da3623..cc473cd0ad6 100644 --- a/python/cugraph-dgl/tests/test_dataloader.py +++ b/python/cugraph-dgl/tests/test_dataloader.py @@ -111,7 +111,6 @@ def test_same_homogeneousgraph_results(): cugraph_output_nodes = cugraph_output[0]["output_nodes"].cpu().numpy() dgl_output_nodes = dgl_output[0]["output_nodes"].cpu().numpy() - np.testing.assert_array_equal(cugraph_output_nodes, dgl_output_nodes) assert ( dgl_output[0]["blocks"][0].num_dst_nodes()