From ce2c1b65ae5c96a1bd424e4a5d55484907a4770b Mon Sep 17 00:00:00 2001
From: Rishabh <59335537+rbSparky@users.noreply.github.com>
Date: Mon, 19 Aug 2024 15:55:49 +0530
Subject: [PATCH] Update conv.jl: Back to old commit

---
 GNNlib/src/layers/conv.jl | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/GNNlib/src/layers/conv.jl b/GNNlib/src/layers/conv.jl
index 2c13e62fc..0b7dd2499 100644
--- a/GNNlib/src/layers/conv.jl
+++ b/GNNlib/src/layers/conv.jl
@@ -355,18 +355,18 @@ end
 
 ####################### MegNetConv ######################################
 
-function megnet_conv(l, g::GNNGraph, x::AbstractMatrix, e::Union{AbstractMatrix, Nothing}=nothing)
+function megnet_conv(l, g::GNNGraph, x::AbstractMatrix, e::AbstractMatrix)
     check_num_nodes(g, x)
-    
-    ē = apply_edges(g, xi = x, xj = x, e = e) do xi, xj, e
+
+    ē = apply_edges(g, xi = x, xj = x, e = e) do xi, xj, e
         l.ϕe(vcat(xi, xj, e))
     end
 
-    xᵉ = aggregate_neighbors(g, l.aggr, ē)
+    xᵉ = aggregate_neighbors(g, l.aggr, ē)
 
     x̄ = l.ϕv(vcat(x, xᵉ))
 
-    return x̄, ē
+    return x̄, ē
 end
 
 ####################### GMMConv ######################################
@@ -721,4 +721,4 @@ function d_conv(l, g::GNNGraph, x::AbstractMatrix)
         T1_out = T2_out
     end
     return h .+ l.bias
-end
\ No newline at end of file
+end