diff --git a/src/ensemble/basic_ensemble_solve.jl b/src/ensemble/basic_ensemble_solve.jl index d32ab1f9bf..bef2d4a5fb 100644 --- a/src/ensemble/basic_ensemble_solve.jl +++ b/src/ensemble/basic_ensemble_solve.jl @@ -29,7 +29,7 @@ function merge_stats(us) reduce(merge, st) end -mutable struct AggregateLogger{T<:AbstractLogger} <: AbstractLogger +mutable struct AggregateLogger{T<:Logging.AbstractLogger} <: Logging.AbstractLogger progress::Dict{Symbol, Float64} done_counter::Int total::Float64 @@ -37,7 +37,7 @@ mutable struct AggregateLogger{T<:AbstractLogger} <: AbstractLogger lock::ReentrantLock logger::T end -AggregateLogger(logger::AbstractLogger) = AggregateLogger(Dict{Symbol, Float64}(),0 , 0.0, 0.0, ReentrantLock(), logger) +AggregateLogger(logger::Logging.AbstractLogger) = AggregateLogger(Dict{Symbol, Float64}(),0 , 0.0, 0.0, ReentrantLock(), logger) function Logging.handle_message(l::AggregateLogger, level, message, _module, group, id, file, line; kwargs...) if convert(LogLevel, level) == LogLevel(-1) && haskey(kwargs, :progress) @@ -130,7 +130,7 @@ function __solve(prob::AbstractEnsembleProblem, ChainRulesCore.@ignore_derivatives if get(kwargs, :progress, false) name = get(kwargs, :progress_name, "Ensemble") for i in 1:trajectories - @logmsg(LogLevel(-1), "$name #$i", _id=Symbol("SciMLBase_$i"), progress=0) + Logging.@logmsg(LogLevel(-1), "$name #$i", _id=Symbol("SciMLBase_$i"), progress=0) end end