Skip to content

Commit

Permalink
Update conv_tests.jl: Fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
rbSparky authored Aug 19, 2024
1 parent ce2c1b6 commit 032e616
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions GNNLux/test/layers/conv_tests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
in_dims = 3
out_dims = 5
x = randn(rng, Float32, in_dims, 10)
"""

@testset "GCNConv" begin
l = GCNConv(in_dims => out_dims, relu)
test_lux_layer(rng, l, g, x, outputsize=(out_dims,))
Expand Down Expand Up @@ -93,9 +93,7 @@
l = GINConv(nn, 0.5)
test_lux_layer(rng, l, g, x, sizey=(out_dims,g.num_nodes), container=true)
end
"""


@testset "MEGNetConv" begin
l = MEGNetConv(in_dims => out_dims)

Expand Down

0 comments on commit 032e616

Please sign in to comment.