From fb0bb1da97f220a2de1dfeb2c368c95ccc746807 Mon Sep 17 00:00:00 2001
From: Rishabh <59335537+rbSparky@users.noreply.github.com>
Date: Sun, 4 Aug 2024 17:16:11 +0530
Subject: [PATCH] Update conv.jl

---
 GNNLux/src/layers/conv.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl
index 009c64cc0..cc838050e 100644
--- a/GNNLux/src/layers/conv.jl
+++ b/GNNLux/src/layers/conv.jl
@@ -656,7 +656,7 @@ function NNConv(ch::Pair{Int, Int}, nn, σ = identity;
     return NNConv(nn, aggr, in_dims, out_dims, use_bias, add_self_loops, use_edge_weight, init_weight, init_bias, σ)
 end
 
-function (l::GCNConv)(g, x, edge_weight, ps, st)
+function (l::NNConv)(g, x, edge_weight, ps, st)
     nn = StatefulLuxLayer{true}(l.nn, ps, st)
 
     # what would be the order of args here?
@@ -669,7 +669,7 @@ end
 
 LuxCore.outputsize(d::NNConv) = (d.out_dims,)
 
-function Base.show(io::IO, l::GINConv)
+function Base.show(io::IO, l::NNConv)
     print(io, "NNConv($(l.nn)")
     print(io, ", $(l.ϵ)")
     l.σ == identity || print(io, ", ", l.σ)