diff --git a/gtsam/discrete/DecisionTreeFactor.h b/gtsam/discrete/DecisionTreeFactor.h index 80ee10a7b..24a699d42 100644 --- a/gtsam/discrete/DecisionTreeFactor.h +++ b/gtsam/discrete/DecisionTreeFactor.h @@ -18,6 +18,7 @@ #pragma once +#include #include #include #include @@ -116,6 +117,10 @@ namespace gtsam { DecisionTreeFactor(const DiscreteKey& key, const std::vector& row) : DecisionTreeFactor(DiscreteKeys{key}, row) {} + /// Construct from Signature + DecisionTreeFactor(const Signature& signature) + : DecisionTreeFactor(signature.discreteKeys(), signature.cpt()) {} + /** Construct from a DiscreteConditional type */ explicit DecisionTreeFactor(const DiscreteConditional& c);