fixup

FluxML · Feb 5, 2022 · f60da1a · f60da1a
1 parent 4dfd551
commit f60da1a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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