Update conv_tests.jl: Fixing tests

JuliaGraphs · Aug 19, 2024 · ff2670c · ff2670c
1 parent 62b3405
commit ff2670c
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/GNNLux/test/layers/conv_tests.jl b/GNNLux/test/layers/conv_tests.jl
@@ -95,14 +95,16 @@
     end
     """
 
+
     @testset "MEGNetConv" begin
         l = MEGNetConv(in_dims => out_dims)
-        
+
         ps = LuxCore.initialparameters(rng, l)
         st = LuxCore.initialstates(rng, l)
-
-        (x_new, e_new), st_new = l(g, x, ps, st)
-
+
+        e = randn(rng, Float32, in_dims, g.num_edges) 
+        (x_new, e_new), st_new = l(g, x, e, ps, st)   
+
         @test size(x_new) == (out_dims, g.num_nodes)
         @test size(e_new) == (out_dims, g.num_edges)
     end