From 834288f9748992b24bc4d4f4cffc77c7d8461d8c Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Fri, 3 Jan 2025 13:18:24 -0500
Subject: [PATCH] additional Signature based constructor for DecisionTreeFactor

---
 gtsam/discrete/DecisionTreeFactor.h | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/gtsam/discrete/DecisionTreeFactor.h b/gtsam/discrete/DecisionTreeFactor.h
index 80ee10a7b..24a699d42 100644
--- a/gtsam/discrete/DecisionTreeFactor.h
+++ b/gtsam/discrete/DecisionTreeFactor.h
@@ -18,6 +18,7 @@
 
 #pragma once
 
+#include <gtsam/discrete/Signature.h>
 #include <gtsam/discrete/AlgebraicDecisionTree.h>
 #include <gtsam/discrete/DiscreteFactor.h>
 #include <gtsam/discrete/DiscreteKey.h>
@@ -116,6 +117,10 @@ namespace gtsam {
     DecisionTreeFactor(const DiscreteKey& key, const std::vector<double>& row)
         : DecisionTreeFactor(DiscreteKeys{key}, row) {}
 
+    /// Construct from Signature
+    DecisionTreeFactor(const Signature& signature)
+        : DecisionTreeFactor(signature.discreteKeys(), signature.cpt()) {}
+
     /** Construct from a DiscreteConditional type */
     explicit DecisionTreeFactor(const DiscreteConditional& c);