fix test

JuliaGraphs · Aug 19, 2024 · fc2db99 · fc2db99
1 parent 890fcda
commit fc2db99
Show file tree

Hide file tree

Showing 4 changed files with 25,164 additions and 3 deletions.
diff --git a/GNNLux/test/layers/basic_tests.jl b/GNNLux/test/layers/basic_tests.jl
@@ -1,6 +1,6 @@
 @testitem "layers/basic" setup=[SharedTestSetup] begin
     rng = StableRNG(17)
-    g = rand_graph(rng, 10, 40)
+    g = rand_graph(10, 40)
     x = randn(rng, Float32, 3, 10)        
 
     @testset "GNNLayer" begin

diff --git a/GNNLux/test/layers/conv_tests.jl b/GNNLux/test/layers/conv_tests.jl
@@ -1,6 +1,6 @@
 @testitem "layers/conv" setup=[SharedTestSetup] begin
     rng = StableRNG(1234)
-    g = rand_graph(rng, 10, 40)
+    g = rand_graph(10, 40)
     in_dims = 3
     out_dims = 5
     x = randn(rng, Float32, in_dims, 10)
@@ -97,7 +97,7 @@
     @testset "NNConv" begin
         edim = 10
         nn = Dense(edim, out_dims * in_dims)
-        l = NNConv(in_dims => out_dims, nn, tanh, bias = true, aggr = +)
+        l = NNConv(in_dims => out_dims, nn, tanh, aggr = +)
         test_lux_layer(rng, l, g, x, sizey=(out_dims, g.num_nodes), container=true)
     end
 end