diff --git a/src/ensemble/ensemble_problems.jl b/src/ensemble/ensemble_problems.jl index c10ea1ff4..fa531098b 100644 --- a/src/ensemble/ensemble_problems.jl +++ b/src/ensemble/ensemble_problems.jl @@ -10,6 +10,14 @@ struct EnsembleProblem{T, T2, T3, T4, T5} <: AbstractEnsembleProblem safetycopy::Bool end +function Base.show(io::IO, t::Type{EnsembleProblem{T, T2, T3, T4, T5}}) where {T, T2, T3, T4, T5} + if TruncatedStacktraces.VERBOSE[] + print(io, "EnsembleProblem{$T,$T2,$T3,$T4,$T5}") + else + print(io, "EnsembleProblem{$T,…}") + end +end + DEFAULT_PROB_FUNC(prob, i, repeat) = prob DEFAULT_OUTPUT_FUNC(sol, i) = (sol, false) DEFAULT_REDUCTION(u, data, I) = append!(u, data), false diff --git a/src/ensemble/ensemble_solutions.jl b/src/ensemble/ensemble_solutions.jl index 10ce3aa9c..b3bfc61ac 100644 --- a/src/ensemble/ensemble_solutions.jl +++ b/src/ensemble/ensemble_solutions.jl @@ -30,6 +30,15 @@ struct EnsembleSolution{T, N, S} <: AbstractEnsembleSolution{T, N, S} elapsedTime::Float64 converged::Bool end + +function Base.show(io::IO, t::Type{EnsembleSolution{T, N, S}}) where {T, N, S} + if TruncatedStacktraces.VERBOSE[] + print(io, "EnsembleSolution{$T,$N,$S}") + else + print(io, "EnsembleProblem{$T,$N,…}") + end +end + function EnsembleSolution(sim, dims::NTuple{N}, elapsedTime, converged) where {N} EnsembleSolution{eltype(eltype(sim)), N, typeof(sim)}(sim, elapsedTime, converged) end