properly handle the case where hop is 0

rapidsai · Dec 2, 2024 · 3a0bf57 · 3a0bf57
1 parent 19c37a8
commit 3a0bf57
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/src/sampling/neighbor_sampling_impl.hpp b/cpp/src/sampling/neighbor_sampling_impl.hpp
@@ -250,7 +250,7 @@ neighbor_sample_impl(raft::handle_t const& handle,
                        starting_vertex_labels,
                        static_cast<size_t>(k_level),
                        with_replacement);
-      } else {
+      } else if (k_level < 0) {
         std::tie(srcs, dsts, weights, edge_ids, edge_types, labels) =
           gather_one_hop_edgelist(handle,
                                   modified_graph_view,