From 84319e5c04659a98b8e5e8d14bc9291170cf0abf Mon Sep 17 00:00:00 2001 From: Roger Waleffe Date: Sat, 28 Oct 2023 12:43:52 -0500 Subject: [PATCH] bug fix --- src/cpp/src/data/samplers/neighbor.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cpp/src/data/samplers/neighbor.cpp b/src/cpp/src/data/samplers/neighbor.cpp index 8403c3b7..7ed6bf5b 100644 --- a/src/cpp/src/data/samplers/neighbor.cpp +++ b/src/cpp/src/data/samplers/neighbor.cpp @@ -496,7 +496,7 @@ DENSEGraph LayeredNeighborSampler::getNeighbors(torch::Tensor node_ids, shared_p } if (outgoing_offsets.defined()) { - if (delta_outgoing_edges.size(0) > 0) { + if (delta_outgoing_offsets.size(0) > 0) { outgoing_offsets = outgoing_offsets + delta_outgoing_edges.size(0); outgoing_offsets = torch::cat({delta_outgoing_offsets, outgoing_offsets}, 0); }