Update conv.jl: Back to old commit

JuliaGraphs · Aug 19, 2024 · ce2c1b6 · ce2c1b6
1 parent ff2670c
commit ce2c1b6
Showing 1 changed file with 6 additions and 6 deletions.
diff --git 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
+
+    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, ē)
 
     x̄ = l.ϕv(vcat(x, xᵉ))
 
-    return x̄, ē
+    return x̄, ē
 end
 
 ####################### GMMConv ######################################
@@ -721,4 +721,4 @@ function d_conv(l, g::GNNGraph, x::AbstractMatrix)
         T1_out = T2_out
     end
     return h .+ l.bias
-end
+end