Skip to content

GitLab

Unverified Commit aa7e3524 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[dev] fix docstring

parent 9b69ff84
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
python/dgl/graphbolt/internal/sample_utils.py
View file @ aa7e3524
...@@ -226,8 +226,8 @@ def unique_and_compact_csc_formats( ...@@ -226,8 +226,8 @@ def unique_and_compact_csc_formats(
... "n1": torch.LongTensor([1, 2]), ... "n1": torch.LongTensor([1, 2]),
... "n2": torch.LongTensor([5, 6])} ... "n2": torch.LongTensor([5, 6])}
>>> csc_formats = { >>> csc_formats = {
... "n1:e1:n2": CSCFormatBase(indptr=torch.tensor([0, 2, 3]),indices=N1), ... "n1:e1:n2": gb.CSCFormatBase(indptr=torch.tensor([0, 2, 3]),indices=N1),
... "n2:e2:n1": CSCFormatBase(indptr=torch.tensor([0, 1, 3]),indices=N2)} ... "n2:e2:n1": gb.CSCFormatBase(indptr=torch.tensor([0, 1, 3]),indices=N2)}
>>> unique_nodes, compacted_csc_formats = gb.unique_and_compact_csc_formats( >>> unique_nodes, compacted_csc_formats = gb.unique_and_compact_csc_formats(
... csc_formats, unique_dst ... csc_formats, unique_dst
... ) ... )
...@@ -340,7 +340,7 @@ def compact_csc_format( ...@@ -340,7 +340,7 @@ def compact_csc_format(
>>> import dgl.graphbolt as gb >>> import dgl.graphbolt as gb
>>> N1 = torch.LongTensor([1, 2, 2]) >>> N1 = torch.LongTensor([1, 2, 2])
>>> N2 = torch.LongTensor([5, 6, 5]) >>> N2 = torch.LongTensor([5, 6, 5])
>>> csc_formats = {"n2:e2:n1": CSCFormatBase(indptr=torch.tensor([0, 1]), >>> csc_formats = {"n2:e2:n1": gb.CSCFormatBase(indptr=torch.tensor([0, 1]),
... indices=torch.tensor([5]))} ... indices=torch.tensor([5]))}
>>> dst_nodes = {"n1": N1[:1]} >>> dst_nodes = {"n1": N1[:1]}
>>> original_row_node_ids, compacted_csc_formats = gb.compact_csc_format( >>> original_row_node_ids, compacted_csc_formats = gb.compact_csc_format(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment