Skip to content

Commit 552c8f2

Browse files
committed
Move src
1 parent 05c5568 commit 552c8f2

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

GNNGraphs/src/GNNGraphs.jl

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,4 +113,7 @@ export mldataset2gnngraph
113113

114114
include("deprecations.jl")
115115

116+
include("samplers.jl")
117+
export NeighborLoader
118+
116119
end #module

GraphNeuralNetworks/src/samplers.jl renamed to GNNGraphs/src/samplers.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,7 @@
33
44
A data structure for sampling neighbors from a graph for training Graph Neural Networks (GNNs).
55
It supports multi-layer sampling of neighbors for a batch of input nodes, useful for mini-batch training
6-
originally introduced in "Inductive Representation Learning on Large Graphs" paper.
7-
[see https://arxiv.org/abs/1706.02216]
6+
originally introduced in ["Inductive Representation Learning on Large Graphs"}(https://arxiv.org/abs/1706.02216) paper.
87
98
# Fields
109
- `graph::GNNGraph`: The input graph.
@@ -21,6 +20,7 @@ julia> batch_counter = 0
2120
julia> for mini_batch_gnn in loader
2221
batch_counter += 1
2322
println("Batch ", batch_counter, ": Nodes in mini-batch graph: ", nv(mini_batch_gnn))
23+
end
2424
```
2525
"""
2626
struct NeighborLoader

GraphNeuralNetworks/src/GraphNeuralNetworks.jl

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,4 @@ export GlobalPool,
6767

6868
include("deprecations.jl")
6969

70-
include("samplers.jl")
71-
export NeighborLoader
72-
7370
end

0 commit comments

Comments
 (0)