From d5cfb7bc4014598a5fa9b47d3e7db60cdca3e798 Mon Sep 17 00:00:00 2001
From: rbSparky <rishb369@gmail.com>
Date: Thu, 8 Aug 2024 04:24:29 +0530
Subject: [PATCH] wip

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

diff --git a/GNNLux/src/GNNLux.jl b/GNNLux/src/GNNLux.jl
index d8970095c..e932451be 100644
--- a/GNNLux/src/GNNLux.jl
+++ b/GNNLux/src/GNNLux.jl
@@ -30,7 +30,7 @@ export AGNNConv,
        GINConv,
        # GMMConv,
        GraphConv,
-       # MEGNetConv,
+       #MEGNetConv,
        # NNConv,
        # ResGatedGraphConv,
        # SAGEConv,
diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl
index 7ea85e0a5..6d7e6f70e 100644
--- a/GNNLux/src/layers/conv.jl
+++ b/GNNLux/src/layers/conv.jl
@@ -650,11 +650,8 @@ function MEGNetConv(ch::Pair{Int, Int}; aggr = mean)
     return MEGNetConv(nin, nout, ϕe, ϕv; aggr)
 end
 
-
 LuxCore.outputsize(l::MegNetConv) = (l.num_features.out,)
 
-(l::MegNetConv)(g, x, ps, st) = l(g, x, nothing, ps, st) # check
-
 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))    
@@ -665,6 +662,6 @@ end
 function Base.show(io::IO, l::MegNetConv)
     nin = l.in_dims
     nout = l.out_dims
-    print(io, "MegNetConv(", l.in_dims, " => ", l.out_dims)
+    print(io, "MegNetConv(", nin, " => ", nout)
     print(io, ")")
 end
\ No newline at end of file