fix testHybridEstimation

release/4.3a0
Varun Agrawal 2025-01-01 21:50:40 -05:00
parent 4c5b842c73
commit e620729c4a
1 changed files with 3 additions and 3 deletions

View File

@ -464,14 +464,14 @@ TEST(HybridEstimation, EliminateSequentialRegression) {
// Create expected discrete conditional on m0.
DiscreteKey m(M(0), 2);
DiscreteConditional expected(m % "0.51341712/1"); // regression
DiscreteTableConditional expected(m % "0.51341712/1"); // regression
// Eliminate into BN using one ordering
const Ordering ordering1{X(0), X(1), M(0)};
HybridBayesNet::shared_ptr bn1 = fg->eliminateSequential(ordering1);
// Check that the discrete conditional matches the expected.
auto dc1 = bn1->back()->asDiscrete();
auto dc1 = bn1->back()->asDiscrete<DiscreteTableConditional>();
EXPECT(assert_equal(expected, *dc1, 1e-9));
// Eliminate into BN using a different ordering
@ -479,7 +479,7 @@ TEST(HybridEstimation, EliminateSequentialRegression) {
HybridBayesNet::shared_ptr bn2 = fg->eliminateSequential(ordering2);
// Check that the discrete conditional matches the expected.
auto dc2 = bn2->back()->asDiscrete();
auto dc2 = bn2->back()->asDiscrete<DiscreteTableConditional>();
EXPECT(assert_equal(expected, *dc2, 1e-9));
}