diff --git a/GNNGraphs/src/sampling.jl b/GNNGraphs/src/sampling.jl index 27e084320..88a7b2fea 100644 --- a/GNNGraphs/src/sampling.jl +++ b/GNNGraphs/src/sampling.jl @@ -58,7 +58,7 @@ function sample_nbrs(loader::NeighborLoader, node::Int, layer::Int) return Int[] else num_samples = min(loader.num_neighbors[layer], length(neighbors)) # Limit to required samples for this layer - return rand(neighbors, num_samples) # Randomly sample neighbors + return StatsBase.sample(neighbors, num_samples; replace = false) # Randomly sample distinct neighbors end end