bug fix

marius-team · Oct 28, 2023 · 84319e5 · 84319e5
1 parent ea44305
commit 84319e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);
             }