From eb2acfaa220ddbbbeb14872d8191af061daefaec Mon Sep 17 00:00:00 2001 From: rbSparky Date: Tue, 19 Mar 2024 16:20:12 +0530 Subject: [PATCH] fix --- src/GNNGraphs/GNNGraphs.jl | 1 + test/GNNGraphs/transform.jl | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/GNNGraphs/GNNGraphs.jl b/src/GNNGraphs/GNNGraphs.jl index 2f989c9b1..1836fa306 100644 --- 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 index ea8ed968e..364e78ac4 100644 --- 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)