fix discrete keys

release/4.3a0
Varun Agrawal 2022-11-03 11:44:41 -04:00
parent 11e4c1ed57
commit 8fa7f44361
1 changed files with 2 additions and 2 deletions

View File

@ -195,7 +195,7 @@ TEST(HybridBayesNet, Error) {
HybridValues delta = hybridBayesNet->optimize();
auto error_tree = hybridBayesNet->error(delta.continuous());
std::vector<DiscreteKey> discrete_keys = {{M(1), 2}, {M(2), 2}};
std::vector<DiscreteKey> discrete_keys = {{M(0), 2}, {M(1), 2}};
std::vector<double> leaves = {0.0097568009, 3.3973404e-31, 0.029126214,
0.0097568009};
AlgebraicDecisionTree<Key> expected_error(discrete_keys, leaves);
@ -216,8 +216,8 @@ TEST(HybridBayesNet, Error) {
// Verify error computation and check for specific error value
DiscreteValues discrete_values;
discrete_values[M(0)] = 1;
discrete_values[M(1)] = 1;
discrete_values[M(2)] = 1;
double total_error = 0;
for (size_t idx = 0; idx < hybridBayesNet->size(); idx++) {