diff --git a/gtsam/hybrid/HybridGaussianFactorGraph.cpp b/gtsam/hybrid/HybridGaussianFactorGraph.cpp index 49f3be776..8e6123f10 100644 --- a/gtsam/hybrid/HybridGaussianFactorGraph.cpp +++ b/gtsam/hybrid/HybridGaussianFactorGraph.cpp @@ -234,13 +234,13 @@ continuousElimination(const HybridGaussianFactorGraph &factors, * and then exponentiate to create a DecisionTreeFactor (not normalized yet!). * * @param errors DecisionTree of (unnormalized) errors. - * @return AlgebraicDecisionTree + * @return DecisionTreeFactor::shared_ptr */ static DecisionTreeFactor::shared_ptr DiscreteFactorFromErrors( const DiscreteKeys &discreteKeys, const AlgebraicDecisionTree &errors) { double min_log = errors.min(); - AlgebraicDecisionTree potentials = DecisionTree( + AlgebraicDecisionTree potentials( errors, [&min_log](const double x) { return exp(-(x - min_log)); }); return std::make_shared(discreteKeys, potentials); }