From 7aeffdbbf58aca82f3d7c7c36e191ac72e8708ea Mon Sep 17 00:00:00 2001 From: bkraske Date: Fri, 21 Jun 2024 10:56:18 -0600 Subject: [PATCH] Actually change all_ba_pruned flag --- src/prune.jl | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/prune.jl b/src/prune.jl index a0ac623..7f19834 100644 --- a/src/prune.jl +++ b/src/prune.jl @@ -34,18 +34,16 @@ function prune!(tree::SARSOPTree) Qa_lower = tree.Qa_lower[b_idx] b_children = tree.b_children[b_idx] max_lower_bound = maximum(Qa_lower) - + all_ba_pruned = true for (idx, Qba) ∈ enumerate(Qa_upper) - all_ba_pruned = true ba_idx = b_children[idx] if !tree.ba_pruned[ba_idx] && Qba < max_lower_bound pruneSubTreeBa!(tree, ba_idx) else all_ba_pruned = false end - all_ba_pruned && (tree.b_pruned[b_idx] = true) end - + all_ba_pruned && (tree.b_pruned[b_idx] = true) end end end