Skip to content

Commit

Permalink
Update conv_tests.jl
Browse files Browse the repository at this point in the history
  • Loading branch information
rbSparky authored Aug 19, 2024
1 parent 24da4c4 commit f2ff073
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion GNNLux/test/layers/conv_tests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -101,6 +101,6 @@
edim = 10
nn = Dense(edim, out_dims * in_dims)
l = NNConv(in_dims => out_dims, nn, tanh, aggr = +)
test_lux_layer(rng, l, g, x, sizey=(out_dims, g.num_nodes), container=true, edge_weight=g.edata.e)
test_lux_layer(rng, l, g2, x, sizey=(out_dims, g2.num_nodes), container=true, edge_weight=g2.edata.e)
end
end

0 comments on commit f2ff073

Please sign in to comment.