fix

CarloLucibello · Mar 19, 2024 · eb2acfa · eb2acfa
1 parent 7b5bc4b
commit eb2acfa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/GNNGraphs/GNNGraphs.jl b/src/GNNGraphs/GNNGraphs.jl
@@ -78,6 +78,7 @@ export add_nodes,
        to_unidirected,
        random_walk_pe,
        remove_nodes,
+       add_edges_augment,
 # from Flux
        batch,
        unbatch,

diff --git a/test/GNNGraphs/transform.jl b/test/GNNGraphs/transform.jl
@@ -157,7 +157,7 @@ end
 
     g = GNNGraph(s,t,eweights)
 
-    g_new = add_edges_randomly(g, weight_func=()->0.8)
+    g_new = add_edges_augment(g, weight_func=()->0.8)
     @test g_new.num_edges == 6
 
     w_new = get_edge_weight(g_new)