import initialobs, fix tests

JuliaPOMDP · Oct 10, 2019 · fde04d9 · fde04d9
1 parent c5ea4da
commit fde04d9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/POMDPModels.jl b/src/POMDPModels.jl
@@ -32,7 +32,7 @@ import POMDPs: gen, support, discount, isterminal
 import POMDPs: actions, actionindex, action, dimensions
 import POMDPs: states, stateindex, transition
 import POMDPs: observations, observation, obsindex
-import POMDPs: initialstate, initialstate_distribution
+import POMDPs: initialstate, initialstate_distribution, initialobs
 import POMDPs: updater, update
 import POMDPs: reward
 import POMDPs: convert_s, convert_a, convert_o

diff --git a/test/tiger.jl b/test/tiger.jl
@@ -23,7 +23,7 @@ let
     simulate(sim, pomdp1, policy, updater(policy), initialstate_distribution(pomdp1))
 
     # test gen(:o, ...)
-    o = gen(DDNNode(:o), pomdp1, true, MersenneTwister(1))
+    o = initialobs(pomdp1, true, MersenneTwister(1))
     @test o == 1
     # test vec
     ov = convert_o(Array{Float64}, true, pomdp1)