From 722b110c7b3d578dae581fd7d5c2723322c53946 Mon Sep 17 00:00:00 2001 From: MaximeBouton Date: Tue, 13 Aug 2019 17:31:18 -0700 Subject: [PATCH] bug fix in action map --- src/sparse.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/sparse.jl b/src/sparse.jl index 010b3bc..6d35210 100644 --- a/src/sparse.jl +++ b/src/sparse.jl @@ -88,7 +88,8 @@ function solve(solver::SparseValueIterationSolver, mdp::SparseTabularMDP) end function solve(solver::SparseValueIterationSolver, mdp::MDP) - return solve(solver, SparseTabularMDP(mdp)) + p = solve(solver, SparseTabularMDP(mdp)) + return ValueIterationPolicy(p.qmat, p.util, p.policy, ordered_actions(mdp), p.include_Q, mdp) end function solve(::SparseValueIterationSolver, ::POMDP)