prune and then add to BayesNet

borglab · Jan 24, 2025 · 2644c4b · 2644c4b
1 parent 26642f1
commit 2644c4b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/gtsam/hybrid/HybridBayesNet.cpp b/gtsam/hybrid/HybridBayesNet.cpp
@@ -58,11 +58,12 @@ HybridBayesNet HybridBayesNet::prune(size_t maxNrLeaves,
     joint = joint * (*conditional);
   }
 
+  // Prune the joint. NOTE: imperative and, again, possibly quite expensive.
+  joint.prune(maxNrLeaves);
+
   // Create the result starting with the pruned joint.
   HybridBayesNet result;
   result.emplace_shared<DiscreteConditional>(joint);
-  // Prune the joint. NOTE: imperative and, again, possibly quite expensive.
-  result.back()->asDiscrete()->prune(maxNrLeaves);
 
   // Get pruned discrete probabilities so
   // we can prune HybridGaussianConditionals.