From cd28e977446729dcee0dfaf40fc2b0a3975ebd55 Mon Sep 17 00:00:00 2001
From: Rishabh <59335537+rbSparky@users.noreply.github.com>
Date: Sun, 4 Aug 2024 17:18:51 +0530
Subject: [PATCH 1/2] Update conv.jl

---
 GNNLux/src/layers/conv.jl | 13 +------------
 1 file changed, 1 insertion(+), 12 deletions(-)

diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl
index da96dfc68..378dd61d6 100644
--- a/GNNLux/src/layers/conv.jl
+++ b/GNNLux/src/layers/conv.jl
@@ -667,17 +667,6 @@ function (l::NNConv)(g, x, edge_weight, ps, st)
     return y, stnew
 end
 
-function LuxCore.initialparameters(rng::AbstractRNG, l::NNConv)
-    weight = l.init_weight(rng, l.out_dims, l.in_dims)
-    if l.use_bias
-        bias = l.init_bias(rng, l.out_dims)
-        return (; weight, bias)
-    else
-        return (; weight)
-    end
-end
-
-LuxCore.parameterlength(l::NNConv) = l.use_bias ? l.in_dims * l.out_dims + l.out_dims : l.in_dims * l.out_dims # nn wont affect this right?
 LuxCore.outputsize(d::NNConv) = (d.out_dims,)
 
 function Base.show(io::IO, l::GINConv)
@@ -688,4 +677,4 @@ function Base.show(io::IO, l::GINConv)
     l.add_self_loops || print(io, ", add_self_loops=false")
     !l.use_edge_weight || print(io, ", use_edge_weight=true")
     print(io, ")")
-end
\ No newline at end of file
+end

From 7f1a07a5f9b7e8d8af54701e822fe53083b4e8d4 Mon Sep 17 00:00:00 2001
From: Rishabh <59335537+rbSparky@users.noreply.github.com>
Date: Sun, 4 Aug 2024 17:19:44 +0530
Subject: [PATCH 2/2] Update conv.jl

---
 GNNLux/src/layers/conv.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl
index 378dd61d6..cc838050e 100644
--- a/GNNLux/src/layers/conv.jl
+++ b/GNNLux/src/layers/conv.jl
@@ -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.σ)