From 27fc6a7a4e3b01cecb65511db36b47fa649a9f79 Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Tue, 13 May 2025 22:49:51 -0400
Subject: [PATCH] add print method to DiscreteMarginals

---
 gtsam/discrete/DiscreteMarginals.cpp | 8 ++++++++
 gtsam/discrete/DiscreteMarginals.h   | 4 ++++
 2 files changed, 12 insertions(+)

diff --git a/gtsam/discrete/DiscreteMarginals.cpp b/gtsam/discrete/DiscreteMarginals.cpp
index d7cc79a5c..74da37e73 100644
--- a/gtsam/discrete/DiscreteMarginals.cpp
+++ b/gtsam/discrete/DiscreteMarginals.cpp
@@ -55,4 +55,12 @@ Vector DiscreteMarginals::marginalProbabilities(const DiscreteKey& key) const {
   return vResult;
 }
 
+/* ************************************************************************* */
+void DiscreteMarginals::print(
+    const std::string& s = "",
+    const KeyFormatter formatter = DefaultKeyFormatter) const {
+  std::cout << (s.empty() ? "Discrete Marginals of:" : s + " ") << std::endl;
+  bayesTree_.print("", formatter);
+}
+
 } /* namespace gtsam */
diff --git a/gtsam/discrete/DiscreteMarginals.h b/gtsam/discrete/DiscreteMarginals.h
index fd6c93fe4..be2e17bcb 100644
--- a/gtsam/discrete/DiscreteMarginals.h
+++ b/gtsam/discrete/DiscreteMarginals.h
@@ -52,6 +52,10 @@ class DiscreteMarginals {
    *   @return Vector of marginal probabilities
    */
   Vector marginalProbabilities(const DiscreteKey& key) const;
+
+  /// Print details
+  void print(const std::string& s = "",
+             const KeyFormatter formatter = DefaultKeyFormatter) const;
 };
 
 } /* namespace gtsam */