fix

JuliaGraphs · Aug 8, 2024 · 5c0c5a8 · 5c0c5a8
1 parent bd2c335
commit 5c0c5a8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl
@@ -637,8 +637,9 @@ end
     aggr::A
 end
 
-# 'mean' not defined
-#MEGNetConv(ϕe, ϕv; aggr = mean) = MEGNetConv(ϕe, ϕv, aggr)
+function MEGNetConv(in_dims::Int, out_dims::Int, ϕe::TE, ϕv::TV; aggr::A = mean) where {TE, TV, A}
+    return MEGNetConv{TE, TV, A}(in_dims, out_dims, ϕe, ϕv, aggr)
+end
 
 function MEGNetConv(ch::Pair{Int, Int}; aggr = mean)
     nin, nout = ch
@@ -648,7 +649,7 @@ function MEGNetConv(ch::Pair{Int, Int}; aggr = mean)
     ϕv = Chain(Dense(nin + nout, nout, relu),
                Dense(nout, nout))
 
-    return MEGNetConv(nin, nout, ϕe, ϕv; aggr)
+    return MEGNetConv(nin, nout, ϕe, ϕv, aggr=aggr)
 end
 
 function (l::MEGNetConv)(g, x, e, ps, st)