diff --git a/src/inference/forward_backward.jl b/src/inference/forward_backward.jl index c36be42..db237ac 100644 --- a/src/inference/forward_backward.jl +++ b/src/inference/forward_backward.jl @@ -38,7 +38,7 @@ function forward_backward!( obs_seq::AbstractVector, control_seq::AbstractVector, seq_ends::AbstractVectorOrNTuple{Int}, - k::Integer, + k::Integer; transition_marginals::Bool=true, ) where {R} (; α, β, c, γ, ξ, B, Bβ) = storage