Update conv_tests.jl: Trying to fix tests

JuliaGraphs · Aug 22, 2024 · f0481b4 · f0481b4
1 parent 3547d9f
commit f0481b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GNNLux/test/layers/conv_tests.jl b/GNNLux/test/layers/conv_tests.jl
@@ -99,7 +99,7 @@
 
     @testset "NNConv" begin
         edim = 10
-        nn = Dense(edim, out_dims * in_dims)        
+        nn = Dense(edim, in_dims * out_dims)        
         l = NNConv(in_dims => out_dims, nn, tanh, aggr = +)
         test_lux_layer(rng, l, g2, x, sizey=(out_dims, g2.num_nodes), container=true, edge_weight=g2.edata.e) 
     end