From c2d3128ef8bb86f1c8144f4e5ab57f1e3deef1c1 Mon Sep 17 00:00:00 2001 From: Agata Skorupka Date: Sun, 6 Oct 2024 22:27:03 +0200 Subject: [PATCH] fix: fix naming for induced_subgraph --- GNNGraphs/src/GNNGraphs.jl | 2 +- GNNGraphs/src/sampling.jl | 2 +- GraphNeuralNetworks/src/GraphNeuralNetworks.jl | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/GNNGraphs/src/GNNGraphs.jl b/GNNGraphs/src/GNNGraphs.jl index d969afd1d..bd87f3c93 100644 --- a/GNNGraphs/src/GNNGraphs.jl +++ b/GNNGraphs/src/GNNGraphs.jl @@ -96,7 +96,7 @@ export rand_graph, rand_temporal_hyperbolic_graph include("sampling.jl") -export sample_neighbors, induced_subgraph +export sample_neighbors, Graphs.induced_subgraph include("operators.jl") # Base.intersect diff --git a/GNNGraphs/src/sampling.jl b/GNNGraphs/src/sampling.jl index d05b560ff..d88790ce4 100644 --- a/GNNGraphs/src/sampling.jl +++ b/GNNGraphs/src/sampling.jl @@ -118,7 +118,7 @@ function sample_neighbors(g::GNNGraph{<:COO_T}, nodes, K = -1; end """ - induced_subgraph(graph::GNNGraph, nodes::Vector{Int}) -> GNNGraph + Graphs.induced_subgraph(graph::GNNGraph, nodes::Vector{Int}) -> GNNGraph Generates a subgraph from the original graph using the provided `nodes`. The function includes the nodes' neighbors and creates edges between nodes that are connected in the original graph. diff --git a/GraphNeuralNetworks/src/GraphNeuralNetworks.jl b/GraphNeuralNetworks/src/GraphNeuralNetworks.jl index 6a8cc0dce..cb6d67104 100644 --- a/GraphNeuralNetworks/src/GraphNeuralNetworks.jl +++ b/GraphNeuralNetworks/src/GraphNeuralNetworks.jl @@ -12,7 +12,7 @@ using Reexport using MLUtils: zeros_like using GNNGraphs: induced_subgraph, - COO_T, ADJMAT_T, SPARSE_T, + COO_T, ADJMAT_T, SPARSE_T,321 check_num_nodes, check_num_edges, EType, NType # for heteroconvs