From ed25af50a399b81c7b7e909ca00f30047b4b2652 Mon Sep 17 00:00:00 2001
From: Mahmoud Bentriou <mahmoud.bentriou@centralesupelec.fr>
Date: Sat, 16 Jan 2021 11:20:30 +0100
Subject: [PATCH] small fix about read_trajectory
---
core/lha.jl | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/core/lha.jl b/core/lha.jl
index c28ad56..ce13e7c 100644
--- a/core/lha.jl
+++ b/core/lha.jl
@@ -268,6 +268,7 @@ end
function read_trajectory(A::LHA, σ::Trajectory; verbose = false)
@assert (σ.m).dim_state == σ.m.dim_obs_state # Model should be entirely obserbed
A_new = LHA(A, (σ.m)._map_obs_var_idx)
+ p_sim = (σ.m).p
l_t = times(σ)
l_tr = transitions(σ)
Sn = init_state(A_new, σ[1], l_t[1])
@@ -275,7 +276,7 @@ function read_trajectory(A::LHA, σ::Trajectory; verbose = false)
if verbose println("Init: ") end
if verbose @show Sn end
for n in 2:length_states(σ)
- next_state!(Snplus1, A_new, σ[n], l_t[n], l_tr[n], Sn, σ[n-1]; verbose = verbose)
+ next_state!(Snplus1, A_new, σ[n], l_t[n], l_tr[n], Sn, σ[n-1], p_sim; verbose = verbose)
copyto!(Sn, Snplus1)
if Snplus1.loc in A_new.locations_final
break
--
GitLab