Skip to content

Commit 894bdb3

Browse files
authored
Update conv_tests.jl
1 parent 39b9c74 commit 894bdb3

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

GNNLux/test/layers/conv_tests.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,13 @@
11
@testitem "layers/conv" setup=[SharedTestSetup] begin
22
rng = StableRNG(1234)
33
edim = 10
4-
g = GNNGraph(g, edata = rand(Float32, edim, g.num_edges))
4+
g = rand_graph(10, 40)
55
in_dims = 3
66
out_dims = 5
77
x = randn(rng, Float32, in_dims, 10)
88

9+
g = GNNGraph(g, edata = rand(Float32, edim, g.num_edges))
10+
911
@testset "GCNConv" begin
1012
l = GCNConv(in_dims => out_dims, tanh)
1113
test_lux_layer(rng, l, g, x, outputsize=(out_dims,))

0 commit comments

Comments
 (0)