From f01bcc0e5933bbf3f4f50d003af9647ce206a608 Mon Sep 17 00:00:00 2001 From: Johannes Fischer Date: Wed, 23 Aug 2023 16:49:59 +0200 Subject: [PATCH] Add generative interface to FullyObservablePOMDP --- lib/POMDPTools/src/ModelTools/fully_observable_pomdp.jl | 1 + 1 file changed, 1 insertion(+) 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)