Skip to content

Commit 048178b

Browse files
committed
Fix default_accumulators internal type
1 parent 6b7b9f8 commit 048178b

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

src/default_accumulators.jl

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -143,10 +143,12 @@ function convert_eltype(::Type{T}, acc::LogLikelihoodAccumulator) where {T}
143143
return LogLikelihoodAccumulator(convert(T, acc.logp))
144144
end
145145

146-
function default_accumulators()
146+
function default_accumulators(
147+
::Type{FloatT}=LogProbType, ::Type{IntT}=Int
148+
) where {FloatT,IntT}
147149
return AccumulatorTuple(
148-
LogPriorAccumulator{LogProbType}(),
149-
LogLikelihoodAccumulator{LogProbType}(),
150-
NumProduceAccumulator{Int}(),
150+
LogPriorAccumulator{FloatT}(),
151+
LogLikelihoodAccumulator{FloatT}(),
152+
NumProduceAccumulator{IntT}(),
151153
)
152154
end

src/simple_varinfo.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ function SimpleVarInfo(values, accs)
204204
return SimpleVarInfo(values, accs, NoTransformation())
205205
end
206206
function SimpleVarInfo{T}(values) where {T<:Real}
207-
return SimpleVarInfo(values, default_accumulators())
207+
return SimpleVarInfo(values, default_accumulators(T))
208208
end
209209
function SimpleVarInfo(values)
210210
return SimpleVarInfo{LogProbType}(values)

0 commit comments

Comments
 (0)