From 578968b6479ddc80d8174ddfa67c41f8a7e0a275 Mon Sep 17 00:00:00 2001
From: rbSparky <rishb369@gmail.com>
Date: Thu, 8 Aug 2024 05:56:51 +0530
Subject: [PATCH] fix

---
 GNNLux/src/layers/conv.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl
index a4fe957e9..9a7259689 100644
--- a/GNNLux/src/layers/conv.jl
+++ b/GNNLux/src/layers/conv.jl
@@ -655,7 +655,7 @@ end
 function (l::MEGNetConv)(g, x, e, ps, st)
     ϕe = StatefulLuxLayer{true}(l.ϕe, ps.ϕe, _getstate(st, :ϕe))
     ϕv = StatefulLuxLayer{true}(l.ϕv, ps.ϕv, _getstate(st, :ϕv))    
-    m = (; ϕe, ϕv)
+    m = (; ϕe, ϕv, l.aggr)
     return GNNlib.megnet_conv(m, g, x, e), st
 end