allow for only continuous variables in HybridBayesTree

release/4.3a0
Varun Agrawal 2025-01-26 12:17:05 -05:00
parent 655f57ef0d
commit 087c0cc525
1 changed files with 5 additions and 0 deletions

View File

@ -202,6 +202,11 @@ VectorValues HybridBayesTree::optimize(const DiscreteValues& assignment) const {
/* ************************************************************************* */
void HybridBayesTree::prune(const size_t maxNrLeaves) {
if (!this->roots_.at(0)->conditional()->asDiscrete()) {
// Root of the BayesTree is not a discrete clique, so we do nothing.
return;
}
auto prunedDiscreteProbs =
this->roots_.at(0)->conditional()->asDiscrete<TableDistribution>();