From b8f5e8f352b0c0deba8035a73ddde4605039a32b Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Sun, 26 Jan 2025 10:31:05 -0500
Subject: [PATCH] undo small change

---
 gtsam/discrete/DecisionTreeFactor.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gtsam/discrete/DecisionTreeFactor.cpp b/gtsam/discrete/DecisionTreeFactor.cpp
index 0396dc952..cfa5fa90c 100644
--- a/gtsam/discrete/DecisionTreeFactor.cpp
+++ b/gtsam/discrete/DecisionTreeFactor.cpp
@@ -453,7 +453,7 @@ namespace gtsam {
   /* ************************************************************************ */
   double DecisionTreeFactor::computeThreshold(const size_t N) const {
     // Set of all keys
-    KeySet allKeys = this->labels();
+    std::set<Key> allKeys = this->labels();
     MinHeap min_heap;
 
     auto op = [&](const Assignment<Key>& a, double p) {