move one line to solve order-of-loading issue

FluxML · Oct 23, 2024 · 3e09cb3 · 3e09cb3
1 parent 0bc7dad
commit 3e09cb3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/layers/basic.jl b/src/layers/basic.jl
@@ -64,6 +64,7 @@ function _applychain(layers::AbstractVector, x)  # type-unstable path, helps com
   end
   x
 end
+_show_pre_post(::Chain{<:AbstractVector}) = "Chain([", "])"  # internal method for show
 
 Base.getindex(c::Chain, i::AbstractArray) = Chain(c.layers[i])
 Base.getindex(c::Chain{<:NamedTuple}, i::AbstractArray) =

diff --git a/src/layers/show.jl b/src/layers/show.jl
@@ -53,7 +53,7 @@ function _big_show(io::IO, obj, indent::Int=0, name=nothing)
 end
 
 _show_pre_post(obj) = string(nameof(typeof(obj)), "("), ")"
-_show_pre_post(::Chain{<:AbstractVector}) = "Chain([", "])"
+# _show_pre_post(::Chain{<:AbstractVector}) = "Chain([", "])"  # has to be done elsewhere
 _show_pre_post(::AbstractVector) = "[", "]"
 _show_pre_post(::NamedTuple) = "(;", ")"