From 53333170791b618f146ec87112d25a1b3ed1acea Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Tue, 13 May 2025 22:50:30 -0400
Subject: [PATCH] wrap DiscreteMarginals class

---
 gtsam/discrete/discrete.i | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

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<gtsam::DiscreteSearchSolution> run(size_t K = 1) const;
 };
 
+#include <gtsam/discrete/DiscreteMarginals.h>
+
+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