Update GNNGraphs/src/transform.jl

Co-authored-by: Carlo Lucibello <[email protected]>
JuliaGraphs · Aug 1, 2024 · b86bf2e · b86bf2e
1 parent 96eb264
commit b86bf2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GNNGraphs/src/transform.jl b/GNNGraphs/src/transform.jl
@@ -151,7 +151,7 @@ end
 
 """
     remove_edges(g::GNNGraph, edges_to_remove::AbstractVector{<:Integer})
-    remove_edges(g::GNNGraph, p::Float64=0.5)
+    remove_edges(g::GNNGraph, p=0.5)
 
 Remove specified edges from a GNNGraph, either by specifying edge indices or by randomly removing edges with a given probability.