Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dgl
Commits
b642c5d8
"src/graph/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "74d88bf832a52b7f37b0b9383a9fdb0b91eda346"
Unverified
Commit
b642c5d8
authored
Aug 29, 2023
by
Rhett Ying
Committed by
GitHub
Aug 29, 2023
Browse files
[GraphBolt] relocate utils (#6227)
parent
2144a3ce
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
5 additions
and
1 deletion
+5
-1
python/dgl/graphbolt/dataloader.py
python/dgl/graphbolt/dataloader.py
+2
-1
python/dgl/graphbolt/utils/__init__.py
python/dgl/graphbolt/utils/__init__.py
+1
-0
python/dgl/graphbolt/utils/datapipe_utils.py
python/dgl/graphbolt/utils/datapipe_utils.py
+2
-0
No files found.
python/dgl/graphbolt/dataloader.py
View file @
b642c5d8
...
@@ -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.
...
...
python/dgl/graphbolt/utils/__init__.py
View file @
b642c5d8
"""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
*
python/dgl/graphbolt/datapipe_utils.py
→
python/dgl/graphbolt/
utils/
datapipe_utils.py
View file @
b642c5d8
"""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
():
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment