#512 fix to avoid dropping some hierarchical addresses from discard d…

…uring update
probcomp · Jun 21, 2024 · 659afba · 659afba
1 parent 95c797f
commit 659afba
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/dynamic/update.jl b/src/dynamic/update.jl
@@ -175,10 +175,11 @@ function add_unvisited_to_discard!(discard::DynamicChoiceMap,
                 set_submap!(discard, key, submap)
             else
                 subdiscard = get_submap(discard, key)
+                subdiscard_recursive = isempty(subdiscard) ? choicemap() : subdiscard
                 add_unvisited_to_discard!(
-                    isempty(subdiscard) ? choicemap() : subdiscard,
+                    subdiscard_recursive,
                     subvisited, submap)
-                set_submap!(discard, key, subdiscard)
+                set_submap!(discard, key, subdiscard_recursive)
             end
         end
     end