From 3d90accef3bae1b0cc93c7facc0e72083378dce4 Mon Sep 17 00:00:00 2001 From: Zachary Sunberg Date: Wed, 27 Dec 2023 18:04:12 -0700 Subject: [PATCH] default_init_state does not need to be generated --- lib/POMDPTools/src/Simulators/sim.jl | 18 ++++++++---------- 1 file changed, 8 insertions(+), 10 deletions(-) diff --git a/lib/POMDPTools/src/Simulators/sim.jl b/lib/POMDPTools/src/Simulators/sim.jl index d2dd271a..1f45d488 100644 --- a/lib/POMDPTools/src/Simulators/sim.jl +++ b/lib/POMDPTools/src/Simulators/sim.jl @@ -125,17 +125,15 @@ function default_init_obs(p::POMDP, s) end end -@generated function default_init_state(p::Union{MDP,POMDP}) - if implemented(initialstate, Tuple{p}) - return :(rand(Random.default_rng(), initialstate(p))) +function default_init_state(m::Union{MDP,POMDP}) + if implemented(initialstate, Tuple{typeof(m)}) + return rand(Random.default_rng(), initialstate(m)) else - return quote - error(""" - Error in sim(::$(typeof(p))): No initial state specified. - - Please supply it as a keyword argument or provide POMDPs.initialstate(::$(typeof(p))). + error(""" + Error in sim(::$(typeof(m))): No initial state specified. + + Please supply it as a keyword argument or provide POMDPs.initialstate(::$(typeof(m))). - """) - end + """) end end