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