From e76ba8347a300d92cf5e53fa6ad69d0a58de7c7e Mon Sep 17 00:00:00 2001
From: Parvm1102 <parvmittal31757@gmail.com>
Date: Sun, 15 Mar 2026 16:39:40 +0530
Subject: [PATCH] fix: sample_nbrs without replacement

Signed-off-by: Parvm1102 <parvmittal31757@gmail.com>
---
 GNNGraphs/src/sampling.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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