diff --git a/gtsam/discrete/discrete.i b/gtsam/discrete/discrete.i index 2f15de4c6..27f1fdfa1 100644 --- a/gtsam/discrete/discrete.i +++ b/gtsam/discrete/discrete.i @@ -494,4 +494,18 @@ class DiscreteSearch { std::vector run(size_t K = 1) const; }; +#include + +class DiscreteMarginals { + DiscreteMarginals(); + DiscreteMarginals(const gtsam::DiscreteFactorGraph& graph); + + gtsam::DiscreteFactor* operator()(gtsam::Key variable) const; + gtsam::Vector marginalProbabilities(const gtsam::DiscreteKey& key) const; + + void print(const std::string& s = "", + const gtsam::KeyFormatter& keyFormatter = + gtsam::DefaultKeyFormatter) const; +}; + } // namespace gtsam