From d5be6d9bcea1e9a67a588bb0b1aab658889d2342 Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Thu, 11 Jul 2024 00:19:17 -0400
Subject: [PATCH] wrap argmax and mode methods

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

diff --git a/gtsam/discrete/discrete.i b/gtsam/discrete/discrete.i
index 5eacb3634..9f55da28e 100644
--- a/gtsam/discrete/discrete.i
+++ b/gtsam/discrete/discrete.i
@@ -137,6 +137,7 @@ virtual class DiscreteConditional : gtsam::DecisionTreeFactor {
   size_t sample(size_t value) const;
   size_t sample() const;
   void sampleInPlace(gtsam::DiscreteValues @parentsValues) const;
+  size_t argmax(const gtsam::DiscreteValues& parents) const;
 
   // Markdown and HTML
   string markdown(const gtsam::KeyFormatter& keyFormatter =
@@ -192,6 +193,7 @@ class DiscreteBayesNet {
 
   gtsam::DiscreteValues sample() const;
   gtsam::DiscreteValues sample(gtsam::DiscreteValues given) const;
+  gtsam::DiscreteValues mode() const;
 
   string dot(
       const gtsam::KeyFormatter& keyFormatter = gtsam::DefaultKeyFormatter,