diff --git a/src/sample.jl b/src/sample.jl index 00535bd..245e82d 100644 --- a/src/sample.jl +++ b/src/sample.jl @@ -12,8 +12,12 @@ function setprogress!(progress::Bool) return progress end -function StatsBase.sample(model_or_logdensity, sampler::AbstractSampler, N_or_isdone; kwargs...) - return StatsBase.sample(Random.default_rng(), model_or_logdensity, sampler, N_or_isdone; kwargs...) +function StatsBase.sample( + model_or_logdensity, sampler::AbstractSampler, N_or_isdone; kwargs... +) + return StatsBase.sample( + Random.default_rng(), model_or_logdensity, sampler, N_or_isdone; kwargs... + ) end """ @@ -72,11 +76,7 @@ Wrap the `logdensity` function in a [`LogDensityModel`](@ref), and call `sample` The `logdensity` function has to support the [LogDensityProblems.jl](https://github.com/tpapp/LogDensityProblems.jl) interface. """ function StatsBase.sample( - rng::Random.AbstractRNG, - logdensity, - sampler::AbstractSampler, - N_or_isdone; - kwargs..., + rng::Random.AbstractRNG, logdensity, sampler::AbstractSampler, N_or_isdone; kwargs... ) return StatsBase.sample(rng, _model(logdensity), sampler, N_or_isdone; kwargs...) end @@ -145,10 +145,11 @@ function StatsBase.sample( nchains::Integer; kwargs..., ) - return StatsBase.sample(rng, _model(logdensity), sampler, parallel, N, nchains; kwargs...) + return StatsBase.sample( + rng, _model(logdensity), sampler, parallel, N, nchains; kwargs... + ) end - # Default implementations of regular and parallel sampling. function mcmcsample( @@ -593,7 +594,11 @@ tighten_eltype(x::Vector{Any}) = map(identity, x) function _model(logdensity) if LogDensityProblems.capabilities(logdensity) === nothing - throw(ArgumentError("the log density function does not support the LogDensityProblems.jl interface. Please implement the interface or provide a model of type `AbstractMCMC.AbstractModel`")) + throw( + ArgumentError( + "the log density function does not support the LogDensityProblems.jl interface. Please implement the interface or provide a model of type `AbstractMCMC.AbstractModel`", + ), + ) end return LogDensityModel(logdensity) end