fix tests

Author: CarloLucibello

src/layers/basic.jl

@@ -64,12 +64,9 @@ applylayer(l, g::GNNGraph, x) = l(x)
 applylayer(l::GNNLayer, g::GNNGraph, x) = l(g, x)
 
 # Handle Flux.Parallel
-applylayer(l::Parallel, g::GNNGraph, x::AbstractArray) = mapreduce(f -> applylayer(l, g, x), l.connection, l.layers)
-applylayer(l::Parallel, g::GNNGraph, xs::Vararg{<:AbstractArray}) = mapreduce((f, x) -> applylayer(l, g, x), l.connection, l.layers, xs)
+applylayer(l::Parallel, g::GNNGraph, x::AbstractArray) = mapreduce(f -> applylayer(f, g, x), l.connection, l.layers)
+applylayer(l::Parallel, g::GNNGraph, xs::Vararg{<:AbstractArray}) = mapreduce((f, x) -> applylayer(f, g, x), l.connection, l.layers, xs)
 applylayer(l::Parallel, g::GNNGraph, xs::Tuple) = applylayer(l, g, xs...)
-applylayer(l::Parallel, g::GNNGraph, x::AbstractArray) = mapreduce(f -> applylayer(l, g, x), l.connection, l.layers)
-applylayer(l::Parallel, g::GNNGraph, xs::Vararg{<:AbstractArray}) = mapreduce((f, x) -> applylayer(l, g, x), l.connection, l.layers, xs)
-
 
 
 applychain(::Tuple{}, g::GNNGraph, x) = x

src/layers/conv.jl

@@ -641,7 +641,7 @@ end
 
 
 function Base.show(io::IO, l::ResGatedGraphConv)
-    out_channel, in_channel = size(l.weight)
+    out_channel, in_channel = size(l.A)
     print(io, "ResGatedGraphConv(", in_channel, "=>", out_channel)
     l.σ == identity || print(io, ", ", l.σ)
     print(io, ")")

test/layers/basic.jl

@@ -15,22 +15,21 @@
 
         testmode!(gnn)
 
-        test_layer(gnn, g, rtol=1e-5) # exclude BN buffers
+        test_layer(gnn, g, rtol=1e-5)
 
 
         @testset "Parallel" begin
             AddResidual(l) = Parallel(+, identity, l) 
 
-            gnn = GNNChain(AddResidual(ResGatedGraphConv(din => d, tanh)),
+            gnn = GNNChain(ResGatedGraphConv(din => d, tanh),
                            BatchNorm(d),
                            AddResidual(ResGatedGraphConv(d => d, tanh)),
                            BatchNorm(d),
                            Dense(d, dout))
 
             testmode!(gnn)
 
-            test_layer(gnn, g, rtol=1e-5, verbose=true, 
-                    exclude_grad_fields=[:μ, :σ², :ϵ]) # exclude BN buffers
+            test_layer(gnn, g, rtol=1e-5)
         end
     end
 end