mean

JuliaGraphs · Aug 8, 2024 · bd2c335 · bd2c335
1 parent 97cc769
commit bd2c335
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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))