Fix

gdalle · Oct 1, 2024 · 01955e8 · 01955e8
1 parent 7aab680
commit 01955e8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/inference/forward.jl b/src/inference/forward.jl
@@ -74,7 +74,7 @@ function forward!(
     seq_ends::AbstractVectorOrNTuple{Int},
     k::Integer,
 )
-    (; α, B, c) = storage
+    (; α, B, c, logL) = storage
     t1, t2 = seq_limits(seq_ends, k)
 
     # Initialization

diff --git a/src/inference/viterbi.jl b/src/inference/viterbi.jl
@@ -47,7 +47,7 @@ function viterbi!(
     seq_ends::AbstractVectorOrNTuple{Int},
     k::Integer,
 ) where {R}
-    (; q, logB, ϕ, ψ) = storage
+    (; q, logB, ϕ, ψ, logL) = storage
     t1, t2 = seq_limits(seq_ends, k)
 
     logBₜ₁ = view(logB, :, t1)