Update conv_tests.jl: test

JuliaGraphs · Sep 14, 2024 · 12ac537 · 12ac537
1 parent 73de0db
commit 12ac537
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
@@ -124,6 +124,6 @@
         ps = LuxCore.initialparameters(rng, l)
         st = LuxCore.initialstates(rng, l)
 
-        y = l(g2, x, e, ps, st) # just to see if it runs without an error
+        test_lux_layer(rng, l, g2, x, sizey=(n_out, g2.num_nodes), container=true, edge_weight=e) 
     end    
 end