fix

JuliaGraphs · Sep 14, 2024 · bbbb07c · bbbb07c
1 parent 12ac537
commit bbbb07c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/GNNLux/test/layers/conv_tests.jl b/GNNLux/test/layers/conv_tests.jl
@@ -124,6 +124,8 @@
         ps = LuxCore.initialparameters(rng, l)
         st = LuxCore.initialstates(rng, l)
 
-        test_lux_layer(rng, l, g2, x, sizey=(n_out, g2.num_nodes), container=true, edge_weight=e) 
+        y, st′ = l(g2, x, e, ps, st)
+
+        @test size(y) == (n_out, g2.num_nodes)
     end    
 end