From 01955e8e6fb4f4b289699e27b1688ad5925c6f95 Mon Sep 17 00:00:00 2001
From: Guillaume Dalle <22795598+gdalle@users.noreply.github.com>
Date: Tue, 1 Oct 2024 13:04:11 +0200
Subject: [PATCH] Fix

---
 src/inference/forward.jl | 2 +-
 src/inference/viterbi.jl | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/inference/forward.jl b/src/inference/forward.jl
index 3e23c8a..653ce63 100644
--- 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
index 070beca..2eb495d 100644
--- 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)