"src/graph/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "74d88bf832a52b7f37b0b9383a9fdb0b91eda346"
Unverified Commit b642c5d8 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[GraphBolt] relocate utils (#6227)

parent 2144a3ce
...@@ -4,10 +4,11 @@ import torch.utils.data ...@@ -4,10 +4,11 @@ import torch.utils.data
import torchdata.dataloader2.graph as dp_utils import torchdata.dataloader2.graph as dp_utils
import torchdata.datapipes as dp import torchdata.datapipes as dp
from .datapipe_utils import datapipe_graph_to_adjlist
from .feature_fetcher import FeatureFetcher from .feature_fetcher import FeatureFetcher
from .minibatch_sampler import MinibatchSampler from .minibatch_sampler import MinibatchSampler
from .utils import datapipe_graph_to_adjlist
class SingleProcessDataLoader(torch.utils.data.DataLoader): class SingleProcessDataLoader(torch.utils.data.DataLoader):
"""Single process DataLoader. """Single process DataLoader.
......
"""Utility functions for GraphBolt.""" """Utility functions for GraphBolt."""
from .internal import * from .internal import *
from .sample_utils import * from .sample_utils import *
from .datapipe_utils import *
"""DataPipe utilities""" """DataPipe utilities"""
__all__ = ["datapipe_graph_to_adjlist"]
def _get_parents(result_dict, datapipe_graph): def _get_parents(result_dict, datapipe_graph):
for k, (v, parents) in datapipe_graph.items(): for k, (v, parents) in datapipe_graph.items():
......
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