Update src/GNNGraphs/transform.jl

Co-authored-by: Carlo Lucibello <[email protected]>
JuliaGraphs · Mar 16, 2024 · ea63a4c · ea63a4c
1 parent 64aad8d
commit ea63a4c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/GNNGraphs/transform.jl b/src/GNNGraphs/transform.jl
@@ -150,7 +150,7 @@ function remove_self_loops(g::GNNGraph{<:ADJMAT_T})
 end
 
 """
-    remove_edges(g::GNNGraph, edges_to_remove::Vector{Int})
+    remove_edges(g::GNNGraph, edges_to_remove::AbstractVector{<:Integer})
 
 Remove specified edges from a GNNGraph.