improve comments

release/4.3a0
Varun Agrawal 2023-06-28 19:09:43 -04:00
parent fceb290e80
commit 64ecb8581e
1 changed files with 2 additions and 1 deletions

View File

@ -154,9 +154,10 @@ void HybridBayesNet::updateDiscreteConditionals(
if (conditional->isDiscrete()) { if (conditional->isDiscrete()) {
auto discrete = conditional->asDiscrete(); auto discrete = conditional->asDiscrete();
// Apply prunerFunc to the underlying AlgebraicDecisionTree // Convert pointer from conditional to factor
auto discreteTree = auto discreteTree =
std::dynamic_pointer_cast<DecisionTreeFactor::ADT>(discrete); std::dynamic_pointer_cast<DecisionTreeFactor::ADT>(discrete);
// Apply prunerFunc to the underlying AlgebraicDecisionTree
DecisionTreeFactor::ADT prunedDiscreteTree = DecisionTreeFactor::ADT prunedDiscreteTree =
discreteTree->apply(prunerFunc(prunedDiscreteProbs, *conditional)); discreteTree->apply(prunerFunc(prunedDiscreteProbs, *conditional));