From e0444ac722d8cb0cc692b17e07d19fc028f713d4 Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Mon, 15 Jul 2024 18:40:07 -0400
Subject: [PATCH] Revert "remove DiscreteLookupTable from wrapper"

This reverts commit ffa72e7fadfff48864c990027f9b60a9f23506e0.
---
 gtsam/discrete/discrete.i | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/gtsam/discrete/discrete.i b/gtsam/discrete/discrete.i
index 0f34840bf..0bdebd0e1 100644
--- a/gtsam/discrete/discrete.i
+++ b/gtsam/discrete/discrete.i
@@ -262,6 +262,15 @@ class DiscreteBayesTree {
 
 #include <gtsam/discrete/DiscreteLookupDAG.h>
 
+class DiscreteLookupTable : gtsam::DiscreteConditional{
+  DiscreteLookupTable(size_t nFrontals, const gtsam::DiscreteKeys& keys,
+                      const gtsam::DecisionTreeFactor::ADT& potentials);
+  void print(string s = "Discrete Lookup Table: ",
+             const gtsam::KeyFormatter& keyFormatter =
+                 gtsam::DefaultKeyFormatter) const;
+  size_t argmax(const gtsam::DiscreteValues& parentsValues) const;
+};
+
 class DiscreteLookupDAG {
   DiscreteLookupDAG();
   void push_back(const gtsam::DiscreteLookupTable* table);