Update conv_tests.jl

JuliaGraphs · Sep 4, 2024 · a587553 · a587553
1 parent 91fed90
commit a587553
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/GNNLux/test/layers/conv_tests.jl b/GNNLux/test/layers/conv_tests.jl
@@ -112,11 +112,11 @@
         l = NNConv(n_in => n_out, nn, tanh, aggr = +)
         x = randn(Float32, n_in, g2.num_nodes)
         e = randn(Float32, n_in_edge, g2.num_edges)
-        y = l(g, x, e) # just to see if it runs without an error
+        #y = l(g, x, e) # just to see if it runs without an error
         #edim = 10
         #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) 
+        test_lux_layer(rng, l, g2, x, sizey=(n_out, g2.num_nodes), container=true, edge_weight=g2.edata.e) 
     end          
 
     @testset "MEGNetConv" begin