fix: fix naming for induced_subgraph

CarloLucibello · Oct 6, 2024 · c2d3128 · c2d3128
1 parent 2ac1ce1
commit c2d3128
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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