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
Show 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(
... "n1": torch.LongTensor([1, 2]),
... "n2": torch.LongTensor([5, 6])}
>>> csc_formats = {
... "n1:e1:n2": CSCFormatBase(indptr=torch.tensor([0, 2, 3]),indices=N1),
... "n2:e2:n1": CSCFormatBase(indptr=torch.tensor([0, 1, 3]),indices=N2)}
... "n1:e1:n2": gb.CSCFormatBase(indptr=torch.tensor([0, 2, 3]),indices=N1),
... "n2:e2:n1": gb.CSCFormatBase(indptr=torch.tensor([0, 1, 3]),indices=N2)}
>>> unique_nodes, compacted_csc_formats = gb.unique_and_compact_csc_formats(
... csc_formats, unique_dst
... )
......@@ -340,7 +340,7 @@ def compact_csc_format(
>>> import dgl.graphbolt as gb
>>> N1 = torch.LongTensor([1, 2, 2])
>>> 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]))}
>>> dst_nodes = {"n1": N1[:1]}
>>> 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