Merge pull request #33 from JuliaPOMDP/sparse

bug fix in action map
JuliaPOMDP · Aug 14, 2019 · 40623ae · 40623ae
2 parents a9df578 + 722b110
commit 40623ae
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)