Skip to content

Commit 00ef0cf

Browse files
committed
Type bound log prob accumulators with T<:Real
1 parent cb1c6c6 commit 00ef0cf

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

src/accumulators.jl

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -193,14 +193,14 @@ end
193193
# END ACCUMULATOR TUPLE, BEGIN LOG PROB AND NUM PRODUCE ACCUMULATORS
194194

195195
"""
196-
LogPriorAccumulator{T} <: AbstractAccumulator
196+
LogPriorAccumulator{T<:Real} <: AbstractAccumulator
197197
198198
An accumulator that tracks the cumulative log prior during model execution.
199199
200200
# Fields
201201
$(TYPEDFIELDS)
202202
"""
203-
struct LogPriorAccumulator{T} <: AbstractAccumulator
203+
struct LogPriorAccumulator{T<:Real} <: AbstractAccumulator
204204
"the scalar log prior value"
205205
logp::T
206206
end
@@ -210,18 +210,18 @@ end
210210
211211
Create a new `LogPriorAccumulator` accumulator with the log prior initialized to zero.
212212
"""
213-
LogPriorAccumulator{T}() where {T} = LogPriorAccumulator(zero(T))
213+
LogPriorAccumulator{T}() where {T<:Real} = LogPriorAccumulator(zero(T))
214214
LogPriorAccumulator() = LogPriorAccumulator{LogProbType}()
215215

216216
"""
217-
LogLikelihoodAccumulator{T} <: AbstractAccumulator
217+
LogLikelihoodAccumulator{T<:Real} <: AbstractAccumulator
218218
219219
An accumulator that tracks the cumulative log likelihood during model execution.
220220
221221
# Fields
222222
$(TYPEDFIELDS)
223223
"""
224-
struct LogLikelihoodAccumulator{T} <: AbstractAccumulator
224+
struct LogLikelihoodAccumulator{T<:Real} <: AbstractAccumulator
225225
"the scalar log likelihood value"
226226
logp::T
227227
end
@@ -231,7 +231,7 @@ end
231231
232232
Create a new `LogLikelihoodAccumulator` accumulator with the log likelihood initialized to zero.
233233
"""
234-
LogLikelihoodAccumulator{T}() where {T} = LogLikelihoodAccumulator(zero(T))
234+
LogLikelihoodAccumulator{T}() where {T<:Real} = LogLikelihoodAccumulator(zero(T))
235235
LogLikelihoodAccumulator() = LogLikelihoodAccumulator{LogProbType}()
236236

237237
"""
@@ -252,7 +252,7 @@ end
252252
253253
Create a new `NumProduceAccumulator` accumulator with the number of observations initialized to zero.
254254
"""
255-
NumProduceAccumulator{T}() where {T} = NumProduceAccumulator(zero(T))
255+
NumProduceAccumulator{T}() where {T<:Integer} = NumProduceAccumulator(zero(T))
256256
NumProduceAccumulator() = NumProduceAccumulator{Int}()
257257

258258
function Base.show(io::IO, acc::LogPriorAccumulator)

0 commit comments

Comments
 (0)