From f60da1af12ab267c6c8895e4ee535529a6a4c06e Mon Sep 17 00:00:00 2001 From: Michael Abbott <32575566+mcabbott@users.noreply.github.com> Date: Sat, 5 Feb 2022 14:36:29 -0500 Subject: [PATCH] fixup --- src/layers/show.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/layers/show.jl b/src/layers/show.jl index 7772b2f49b..a37af36065 100644 --- a/src/layers/show.jl +++ b/src/layers/show.jl @@ -102,7 +102,7 @@ function _big_finale(io::IO, m) end end -_childarray_sum(f, x::AbstractArray) = f(x) +_childarray_sum(f, x::AbstractArray{<:Number}) = f(x) _childarray_sum(f, x) = isleaf(x) ? 0 : sum(y -> _childarray_sum(f, y), Functors.children(x)) # utility functions