From be04df5a3421308c57d14f758742108bd7dfbb8a Mon Sep 17 00:00:00 2001
From: Rishabh <59335537+rbSparky@users.noreply.github.com>
Date: Mon, 30 Sep 2024 14:32:47 +0530
Subject: [PATCH] Update conv.jl: self loops

---
 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 484c5d9a2..d1353b9c2 100644
--- a/GNNLux/src/layers/conv.jl
+++ b/GNNLux/src/layers/conv.jl
@@ -929,11 +929,11 @@ function (l::TransformerConv)(g, x, e, ps, st)
     BN2 = l.BN2 === nothing ? nothing : 
             StatefulLuxLayer{true}(l.BN2, ps.BN2, _getstate(st, :BN2))
     m = (; W1, W2, W3, W4, W5, W6, FF, BN1, BN2, l.sqrt_out, 
-           l.heads, l.concat, l.skip_connection)
+           l.heads, l.concat, l.skip_connection, l.add_self_loops)
     return GNNlib.transformer_conv(m, g, x, e), st
 end
 
 function Base.show(io::IO, l::TransformerConv)
     (in, ein), out = (l.in_dims, l.out_dims)
     print(io, "TransformerConv(($in, $ein) => $out, heads=$(l.heads))")
-end
\ No newline at end of file
+end