Skip to content

GitLab

"tutorials/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "7e923180c17c33bbe51c22c883cc67a64934da5f"
Unverified Commit 72937574 authored by Mingbang Wang's avatar Mingbang Wang Committed by GitHub
Browse files

[Misc] Polish the python code (#6704)

parent 130d84a2
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 1 deletion
python/dgl/graphbolt/impl/fused_csc_sampling_graph.py
View file @ 72937574
...@@ -294,7 +294,7 @@ class FusedCSCSamplingGraph(SamplingGraph): ...@@ -294,7 +294,7 @@ class FusedCSCSamplingGraph(SamplingGraph):
Returns Returns
------- -------
torch.Tensor or None Dict[str, torch.Tensor] or None
If present, returns a dictionary of edge attributes. Each key If present, returns a dictionary of edge attributes. Each key
represents the attribute's name, while the corresponding value represents the attribute's name, while the corresponding value
holds the attribute's specific value. The length of each value holds the attribute's specific value. The length of each value
...@@ -583,6 +583,7 @@ class FusedCSCSamplingGraph(SamplingGraph): ...@@ -583,6 +583,7 @@ class FusedCSCSamplingGraph(SamplingGraph):
corresponding to each neighboring edge of a node. It must be a 1D corresponding to each neighboring edge of a node. It must be a 1D
floating-point or boolean tensor, with the number of elements floating-point or boolean tensor, with the number of elements
equalling the total number of edges. equalling the total number of edges.
Returns Returns
------- -------
FusedSampledSubgraphImpl FusedSampledSubgraphImpl
...@@ -698,6 +699,7 @@ class FusedCSCSamplingGraph(SamplingGraph): ...@@ -698,6 +699,7 @@ class FusedCSCSamplingGraph(SamplingGraph):
corresponding to each neighboring edge of a node. It must be a 1D corresponding to each neighboring edge of a node. It must be a 1D
floating-point or boolean tensor, with the number of elements floating-point or boolean tensor, with the number of elements
equalling the total number of edges. equalling the total number of edges.
Returns Returns
------- -------
torch.classes.graphbolt.SampledSubgraph torch.classes.graphbolt.SampledSubgraph
...@@ -767,6 +769,7 @@ class FusedCSCSamplingGraph(SamplingGraph): ...@@ -767,6 +769,7 @@ class FusedCSCSamplingGraph(SamplingGraph):
corresponding to each neighboring edge of a node. It must be a 1D corresponding to each neighboring edge of a node. It must be a 1D
floating-point or boolean tensor, with the number of elements floating-point or boolean tensor, with the number of elements
equalling the total number of edges. equalling the total number of edges.
Returns Returns
------- -------
FusedSampledSubgraphImpl FusedSampledSubgraphImpl
...@@ -934,6 +937,7 @@ def from_fused_csc( ...@@ -934,6 +937,7 @@ def from_fused_csc(
Edge attributes of the graph, by default None. Edge attributes of the graph, by default None.
metadata: Optional[GraphMetadata], optional metadata: Optional[GraphMetadata], optional
Metadata of the graph, by default None. Metadata of the graph, by default None.
Returns Returns
------- -------
FusedCSCSamplingGraph FusedCSCSamplingGraph
......
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