diff --git a/src/tracker/lib/array.jl b/src/tracker/lib/array.jl index 545cbb99..1bef3bb7 100644 --- a/src/tracker/lib/array.jl +++ b/src/tracker/lib/array.jl @@ -53,6 +53,11 @@ end Base.print_array(io::IO, x::TrackedArray) = Base.print_array(io, data(x)) +function Base.show(io::IO, x::TrackedArray) + show(io, data(x)) + print(io, " (tracked)") +end + Base.copy(x::TrackedArray) = x Base.setindex!(xs::TrackedArray, v, i...) = diff --git a/src/tracker/lib/real.jl b/src/tracker/lib/real.jl index e494b8ef..c5acf9fe 100644 --- a/src/tracker/lib/real.jl +++ b/src/tracker/lib/real.jl @@ -17,8 +17,9 @@ function back!(x::TrackedReal; once = true) end function Base.show(io::IO, x::TrackedReal) + T = get(io, :typeinfo, Any) show(io, data(x)) - print(io, " (tracked)") + T <: TrackedReal || print(io, " (tracked)") end Base.decompose(x::TrackedReal) = Base.decompose(data(x))