Skip to content

GitLab

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

[GraphBolt] fix docstring (#6502)

parent 23c566a6
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
python/dgl/graphbolt/impl/ondisk_dataset.py
View file @ 1d86b796
...@@ -44,6 +44,8 @@ def preprocess_ondisk_dataset( ...@@ -44,6 +44,8 @@ def preprocess_ondisk_dataset(
---------- ----------
dataset_dir : str dataset_dir : str
The path to the dataset directory. The path to the dataset directory.
include_original_edge_id : bool, optional
Whether to include the original edge id in the CSCSamplingGraph.
Returns Returns
------- -------
...@@ -337,6 +339,8 @@ class OnDiskDataset(Dataset): ...@@ -337,6 +339,8 @@ class OnDiskDataset(Dataset):
---------- ----------
path: str path: str
The YAML file path. The YAML file path.
include_original_edge_id: bool, optional
Whether to include the original edge id in the CSCSamplingGraph.
""" """
def __init__( def __init__(
......
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