From 8255ad596cecda7e9c9a6ee050d68d13df25c83c Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Mon, 18 Sep 2023 22:57:24 -0400
Subject: [PATCH] add Testable traits to DiscreteBayesTree and HybridBayesTree

---
 gtsam/discrete/DiscreteBayesTree.h | 8 ++++++++
 gtsam/hybrid/HybridBayesTree.h     | 4 ++++
 2 files changed, 12 insertions(+)

diff --git a/gtsam/discrete/DiscreteBayesTree.h b/gtsam/discrete/DiscreteBayesTree.h
index 34f8f5d4a..89baf49a9 100644
--- a/gtsam/discrete/DiscreteBayesTree.h
+++ b/gtsam/discrete/DiscreteBayesTree.h
@@ -110,4 +110,12 @@ class GTSAM_EXPORT DiscreteBayesTree
   /// @}
 };
 
+/// traits
+template <>
+struct traits<DiscreteBayesTreeClique>
+    : public Testable<DiscreteBayesTreeClique> {};
+
+template <>
+struct traits<DiscreteBayesTree> : public Testable<DiscreteBayesTree> {};
+
 }  // namespace gtsam
diff --git a/gtsam/hybrid/HybridBayesTree.h b/gtsam/hybrid/HybridBayesTree.h
index bc2a0f852..f91e16cbf 100644
--- a/gtsam/hybrid/HybridBayesTree.h
+++ b/gtsam/hybrid/HybridBayesTree.h
@@ -123,6 +123,10 @@ class GTSAM_EXPORT HybridBayesTree : public BayesTree<HybridBayesTreeClique> {
 };
 
 /// traits
+template <>
+struct traits<HybridBayesTreeClique> : public Testable<HybridBayesTreeClique> {
+};
+
 template <>
 struct traits<HybridBayesTree> : public Testable<HybridBayesTree> {};