small modification to profiling

JuliaPOMDP · Jan 19, 2025 · bf278c1 · bf278c1
1 parent 6d9c943
commit bf278c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/profile/pomcp.jl b/profile/pomcp.jl
@@ -1,4 +1,4 @@
-using POMDPs, POMDPModels, BasicPOMCP, ParticleFilters, ProfileView
+using POMDPs, POMDPModels, BasicPOMCP, ParticleFilters
 
 m = TigerPOMDP()
 solver = POMCPSolver(tree_queries=10_000, c=100.0)
@@ -10,5 +10,5 @@ a = action(planner, b)
 
 b = initialize_belief(pf, initialstate(m))
 @profview for i in 1:100
-    action(planner, b)
+    action(planner, initialize_belief(pf, initialstate(m)))
 end