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 Sep 4, 2024
1 parent a587553 commit 232a1b4
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 @@ -116,7 +116,7 @@
#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=(n_out, 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=e)
end

@testset "MEGNetConv" begin
Expand Down

0 comments on commit 232a1b4

Please sign in to comment.