From bd2c3351551533e4fa1ed77724660af074952bd1 Mon Sep 17 00:00:00 2001
From: rbSparky <rishb369@gmail.com>
Date: Thu, 8 Aug 2024 05:38:00 +0530
Subject: [PATCH] mean

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

diff --git a/GNNLux/src/GNNLux.jl b/GNNLux/src/GNNLux.jl
index 3aa3251d0..b24ed4118 100644
--- a/GNNLux/src/GNNLux.jl
+++ b/GNNLux/src/GNNLux.jl
@@ -1,6 +1,7 @@
 module GNNLux
 using ConcreteStructs: @concrete
 using NNlib: NNlib, sigmoid, relu, swish
+using Statistics: mean
 using LuxCore: LuxCore, AbstractExplicitLayer, AbstractExplicitContainerLayer, parameterlength, statelength, outputsize, 
               initialparameters, initialstates, parameterlength, statelength
 using Lux: Lux, Chain, Dense, GRUCell,
diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl
index b05f46020..e66cdfcbd 100644
--- a/GNNLux/src/layers/conv.jl
+++ b/GNNLux/src/layers/conv.jl
@@ -640,7 +640,7 @@ end
 # 'mean' not defined
 #MEGNetConv(ϕe, ϕv; aggr = mean) = MEGNetConv(ϕe, ϕv, aggr)
 
-function MEGNetConv(ch::Pair{Int, Int}; aggr)
+function MEGNetConv(ch::Pair{Int, Int}; aggr = mean)
     nin, nout = ch
     ϕe = Chain(Dense(3nin, nout, relu),
                Dense(nout, nout))