Update src/GNNGraphs/transform.jl

Co-authored-by: Carlo Lucibello <[email protected]>
JuliaGraphs · Jul 4, 2024 · 0d3017d · 0d3017d
1 parent c04a832
commit 0d3017d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/GNNGraphs/transform.jl b/src/GNNGraphs/transform.jl
@@ -413,7 +413,7 @@ function add_edges(g::GNNHeteroGraph{<:COO_T},
 end
 
 """
-    perturb_edges(g::GNNGraph, perturb_ratio; [seed])
+    perturb_edges([rng], g::GNNGraph, perturb_ratio)
 
 Perturb the graph `g` by adding random edges, based on a specified `perturb_ratio`. The `perturb_ratio` determines the fraction of new edges to add relative to the current number of edges in the graph. These new edges are added without creating self-loops. Optionally, a random `seed` can be provided to ensure reproducible perturbations.