Add generative interface to FullyObservablePOMDP

JuliaPOMDP · Aug 23, 2023 · f01bcc0 · f01bcc0
1 parent ec89812
commit f01bcc0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lib/POMDPTools/src/ModelTools/fully_observable_pomdp.jl b/lib/POMDPTools/src/ModelTools/fully_observable_pomdp.jl
@@ -26,6 +26,7 @@ POMDPs.observation(::FullyObservablePOMDP, a, sp) = Deterministic(sp)
 POMDPs.states(pomdp::FullyObservablePOMDP) = states(pomdp.mdp)
 POMDPs.actions(pomdp::FullyObservablePOMDP) = actions(pomdp.mdp)
 POMDPs.transition(pomdp::FullyObservablePOMDP, s, a) = transition(pomdp.mdp, s, a)
+POMDPs.gen(pomdp::FullyObservablePOMDP, s, a, rng) = gen(pomdp.mdp, s, a, rng)
 POMDPs.isterminal(pomdp::FullyObservablePOMDP, s) = isterminal(pomdp.mdp, s)
 POMDPs.discount(pomdp::FullyObservablePOMDP) = discount(pomdp.mdp)
 POMDPs.stateindex(pomdp::FullyObservablePOMDP, s) = stateindex(pomdp.mdp, s)