fix test_dataset test

rapidsai · Sep 12, 2023 · b1f8ff2 · b1f8ff2
1 parent bc8e141
commit b1f8ff2
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/python/cugraph-dgl/tests/test_dataset.py b/python/cugraph-dgl/tests/test_dataset.py
@@ -123,6 +123,6 @@ def test_homogeneous_sampled_graphs_from_dataframe(return_type, seed_node):
             assert dgl_block.num_src_nodes() == cugraph_dgl_graph.num_src_nodes()
             assert dgl_block.num_dst_nodes() == cugraph_dgl_graph.num_dst_nodes()
             dgl_offsets, dgl_indices, _ = dgl_block.adj_tensors("csc")
-            cugraph_offsets, cugraph_indices = cugraph_dgl_graph.csc()
+            cugraph_offsets, cugraph_indices, _ = cugraph_dgl_graph.csc()
             assert torch.equal(dgl_offsets.to("cpu"), cugraph_offsets.to("cpu"))
             assert torch.equal(dgl_indices.to("cpu"), cugraph_indices.to("cpu"))
diff --git a/...aph-dgl/tests/test_from_dgl_hetrograph.py → ...ph-dgl/tests/test_from_dgl_heterograph.py b/...aph-dgl/tests/test_from_dgl_hetrograph.py → ...ph-dgl/tests/test_from_dgl_heterograph.py