Be cautious and print retcode only if available

SciML · Nov 24, 2023 · 912f3ca · 912f3ca
1 parent 45fa653
commit 912f3ca
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/solutions/nonlinear_solutions.jl b/src/solutions/nonlinear_solutions.jl
@@ -93,8 +93,11 @@ function sensitivity_solution(sol::AbstractNonlinearSolution, u)
     T = eltype(eltype(u))
     N = ndims(u)
 
+    # Some of the subtypes might not have a trace field
+    trace = hasfield(typeof(sol), :trace) ? sol.trace : nothing
+
     NonlinearSolution{T, N, typeof(u), typeof(sol.resid), typeof(sol.prob),
         typeof(sol.alg), typeof(sol.original), typeof(sol.left),
-        typeof(sol.stats), trace(sol.trace)}(u, sol.resid, sol.prob, sol.alg, sol.retcode,
-        sol.original, sol.left, sol.right, sol.stats, sol.trace)
+        typeof(sol.stats), trace(trace)}(u, sol.resid, sol.prob, sol.alg, sol.retcode,
+        sol.original, sol.left, sol.right, sol.stats, trace)
 end
diff --git a/src/solutions/solution_interface.jl b/src/solutions/solution_interface.jl
@@ -14,7 +14,9 @@ Base.setindex!(A::AbstractNoTimeSolution, v, I::Vararg{Int, N}) where {N} = (A.u
 Base.size(A::AbstractNoTimeSolution) = size(A.u)
 
 function Base.show(io::IO, m::MIME"text/plain", A::AbstractNoTimeSolution)
-    println(io, string("retcode: ", A.retcode))
+    if hasfield(typeof(A), :retcode)
+        println(io, string("retcode: ", A.retcode))
+    end
     print(io, "u: ")
     show(io, m, A.u)
 end