fix output

JuliaGraphs · Aug 4, 2024 · 4904f91 · 4904f91
1 parent 9b76cf5
commit 4904f91
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl
@@ -663,9 +663,8 @@ function (l::MegNetConv)(g, x, e, ps, st)
 end
 
 function Base.show(io::IO, l::MegNetConv)
-    ne = l.num_features.edge
-    nin = l.num_features.in
-    nout = l.num_features.out
-    print(io, "MegNetConv(($nin, $ne) => $nout")
+    nin = l.in_dims
+    nout = l.out_dims
+    print(io, "MegNetConv(", l.in_dims, " => ", l.out_dims)
     print(io, ")")
 end