12345qiupeng
/
gtsam
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix pruning test in testHybridBayesNet

release/4.3a0
Varun Agrawal 2025-01-01 22:06:06 -05:00
parent da22055f35
commit fcc56f5de6
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
gtsam/hybrid/tests/testHybridBayesNet.cpp
View File

@ -450,7 +450,15 @@ TEST(HybridBayesNet, UpdateDiscreteConditionals) {
DiscreteConditional joint;
for (auto&& conditional : posterior->discreteMarginal()) {
joint = joint * (*conditional);
// The last discrete conditional may be a DiscreteTableConditional
if (auto dtc =
std::dynamic_pointer_cast<DiscreteTableConditional>(conditional)) {
DiscreteConditional dc(dtc->nrFrontals(),
dtc->table().toDecisionTreeFactor());
joint = joint * dc;
} else {
joint = joint * (*conditional);
}
}
size_t maxNrLeaves = 3;