fix stage(w, ss) call

Omastto1 · Apr 19, 2021 · d4d123f · d4d123f
1 parent bad7680
commit d4d123f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/fixhorizon.jl b/src/fixhorizon.jl
@@ -119,7 +119,7 @@ end
 Create a product of Infinite Horizon MDP's observations given destination state and action (and original state) with original state's stage.
 """
 POMDPs.observation(w::FixedHorizonPOMDPWrapper, ss::Tuple{<:Any,Int}, a, ssp::Tuple{<:Any, Int}) = InStageDistribution(observation(w.m, first(ss), a, first(ssp)), stage(w, ss))
-POMDPs.observation(w::FixedHorizonPOMDPWrapper, a, ssp::Tuple{<:Any, Int}) = InStageDistribution(observation(w.m, a, first(ssp)), stage(ssp)-1)
+POMDPs.observation(w::FixedHorizonPOMDPWrapper, a, ssp::Tuple{<:Any, Int}) = InStageDistribution(observation(w.m, a, first(ssp)), stage(w, ssp)-1)
 
 """
     POMDPs.initialstate(w::FHWrapper)