From 14f32544d26c685d861233070901fb5955768c82 Mon Sep 17 00:00:00 2001 From: Varun Agrawal Date: Sat, 4 Jan 2025 14:39:53 -0500 Subject: [PATCH] update test --- gtsam/hybrid/tests/testHybridBayesNet.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/gtsam/hybrid/tests/testHybridBayesNet.cpp b/gtsam/hybrid/tests/testHybridBayesNet.cpp index 247474c6b..5c788446c 100644 --- a/gtsam/hybrid/tests/testHybridBayesNet.cpp +++ b/gtsam/hybrid/tests/testHybridBayesNet.cpp @@ -451,8 +451,7 @@ TEST(HybridBayesNet, UpdateDiscreteConditionals) { DiscreteConditional joint; for (auto&& conditional : posterior->discreteMarginal()) { // The last discrete conditional may be a TableDistribution - if (auto dtc = - std::dynamic_pointer_cast(conditional)) { + if (auto dtc = std::dynamic_pointer_cast(conditional)) { DiscreteConditional dc(dtc->nrFrontals(), dtc->toDecisionTreeFactor()); joint = joint * dc; } else { @@ -461,7 +460,8 @@ TEST(HybridBayesNet, UpdateDiscreteConditionals) { } size_t maxNrLeaves = 3; - auto prunedDecisionTree = *joint.prune(maxNrLeaves); + DiscreteConditional prunedDecisionTree(joint); + prunedDecisionTree.prune(maxNrLeaves); #ifdef GTSAM_DT_MERGING EXPECT_LONGS_EQUAL(maxNrLeaves + 2 /*2 zero leaves*/,