From 894bdb31293ad35a0003c41b645819345d93cecf Mon Sep 17 00:00:00 2001 From: Rishabh <59335537+rbSparky@users.noreply.github.com> Date: Mon, 19 Aug 2024 16:09:10 +0530 Subject: [PATCH] Update conv_tests.jl --- GNNLux/test/layers/conv_tests.jl | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/GNNLux/test/layers/conv_tests.jl b/GNNLux/test/layers/conv_tests.jl index e6db4e130..0c743a048 100644 --- a/GNNLux/test/layers/conv_tests.jl +++ b/GNNLux/test/layers/conv_tests.jl @@ -1,11 +1,13 @@ @testitem "layers/conv" setup=[SharedTestSetup] begin rng = StableRNG(1234) edim = 10 - g = GNNGraph(g, edata = rand(Float32, edim, g.num_edges)) + g = rand_graph(10, 40) in_dims = 3 out_dims = 5 x = randn(rng, Float32, in_dims, 10) + g = GNNGraph(g, edata = rand(Float32, edim, g.num_edges)) + @testset "GCNConv" begin l = GCNConv(in_dims => out_dims, tanh) test_lux_layer(rng, l, g, x, outputsize=(out_dims,))