Skip to content

Commit

Permalink
clarify the docstring more.
Browse files Browse the repository at this point in the history
  • Loading branch information
mfbalin committed Aug 26, 2024
1 parent 2ced908 commit 1f021e2
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion python/dgl/graphbolt/sampled_subgraph.py
Original file line number Diff line number Diff line change
Expand Up @@ -270,7 +270,8 @@ def to_pyg(
-------
Union[PyGLayerData, PyGLayerHeteroData]
A named tuple class with `x`, `edge_index` and `size` fields.
Typically, a PyG GNN layer will accept these parameters.
Typically, a PyG GNN layer's forward method will accept these as
arguments.
"""
if isinstance(x, torch.Tensor):
# Homogenous
Expand Down

0 comments on commit 1f021e2

Please sign in to comment.