shorten leaf printing

FluxML · Feb 5, 2022 · c198f70 · c198f70
1 parent adc0e85
commit c198f70
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/interface.jl b/src/interface.jl
@@ -116,8 +116,11 @@ onevalue(λ, x::AbstractArray{T}) where T = onevalue(convert(float(T), λ), x)
 
 function Base.show(io::IO, ℓ::Leaf)  # show method is mostly to hide its long type!
   ioc = IOContext(io, :compact => true)
-  print(ioc, "Leaf(", ℓ.rule, ", ")
-  show(ioc, ℓ.state)
-  print(io, ")")
+  printstyled(io, "Leaf("; color = :green)
+  show(ioc, ℓ.rule)
+  printstyled(", "; color = :green)
+  str = sprint(show, ℓ.state; context = ioc)
+  print(io, length(str) < 70 ? str : first(str, 50) * " … ")
+  printstyled(io, ")"; color = :green)
 end