From d4d123fcff3a4618f54a57c8cdfbc6a09e070b22 Mon Sep 17 00:00:00 2001 From: Tomas Omasta Date: Mon, 19 Apr 2021 23:17:22 +0200 Subject: [PATCH] fix stage(w, ss) call --- src/fixhorizon.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/fixhorizon.jl b/src/fixhorizon.jl index 81dc661..ccbc5f0 100644 --- 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)