Skip to content

Commit 4f1dd75

Browse files
committed
Add display method for ADResult too
1 parent b54a554 commit 4f1dd75

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

src/test_utils/ad.jl

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -168,6 +168,19 @@ struct ADResult{Tparams<:AbstractFloat,Tresult<:AbstractFloat,Ttol<:AbstractFloa
168168
seconds)"
169169
primal_time::Union{Nothing,Tresult}
170170
end
171+
function Base.show(io::IO, ::MIME"text/plain", result::ADResult)
172+
println(io, "ADResult for $(result.model.f) with $(result.adtype):")
173+
println(io, " ├ value_actual : $(result.value_actual)")
174+
println(io, " ├ value_expected : $(result.value_expected)")
175+
println(io, " ├ grad_actual : $(result.grad_actual)")
176+
println(io, " ├ grad_expected : $(result.grad_expected)")
177+
if result.grad_time !== nothing && result.primal_time !== nothing
178+
println(io, " ├ grad_time : $(result.grad_time) s")
179+
println(io, " ├ primal_time : $(result.primal_time) s")
180+
end
181+
println(io, " ├ model : $(result.model.f)")
182+
return println(io, " └ params : $(result.params)")
183+
end
171184

172185
"""
173186
run_ad(

0 commit comments

Comments
 (0)