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 39b9c74 commit 894bdb3
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion GNNLux/test/layers/conv_tests.jl
Original file line number Diff line number Diff line change
@@ -1,11 +1,13 @@
@testitem "layers/conv" setup=[SharedTestSetup] begin
rng = StableRNG(1234)
edim = 10
g = GNNGraph(g, edata = rand(Float32, edim, g.num_edges))
g = rand_graph(10, 40)
in_dims = 3
out_dims = 5
x = randn(rng, Float32, in_dims, 10)

g = GNNGraph(g, edata = rand(Float32, edim, g.num_edges))

@testset "GCNConv" begin
l = GCNConv(in_dims => out_dims, tanh)
test_lux_layer(rng, l, g, x, outputsize=(out_dims,))
Expand Down

0 comments on commit 894bdb3

Please sign in to comment.