diff --git a/lib/POMDPTools/src/ModelTools/fully_observable_pomdp.jl b/lib/POMDPTools/src/ModelTools/fully_observable_pomdp.jl index 0f994cbc..e42fa221 100644 --- 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)