Update shared_testsetup.jl: fixing other tests

JuliaGraphs · Aug 25, 2024 · b1d185f · b1d185f
1 parent 6227cd3
commit b1d185f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/GNNLux/test/shared_testsetup.jl b/GNNLux/test/shared_testsetup.jl
@@ -28,9 +28,9 @@ function test_lux_layer(rng::AbstractRNG, l, g::GNNGraph, x;
     @test LuxCore.statelength(l) == LuxCore.statelength(st)
 
     if edge_weight !== nothing
-        y, st′ = l(g, x, ps, st)
-    else          
         y, st′ = l(g, x, edge_weight, ps, st)
+    else          
+        y, st′ = l(g, x, ps, st)
     end
 
     @test eltype(y) == eltype(x)