From b1d185fdd1f8510996077b9f9acee5cb075b4f0c Mon Sep 17 00:00:00 2001
From: Rishabh <59335537+rbSparky@users.noreply.github.com>
Date: Mon, 26 Aug 2024 03:51:50 +0530
Subject: [PATCH] Update shared_testsetup.jl: fixing other tests

---
 GNNLux/test/shared_testsetup.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/GNNLux/test/shared_testsetup.jl b/GNNLux/test/shared_testsetup.jl
index bcd243df3..797e1577d 100644
--- 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)