From f4ea5511535b21b203d9c540bceb4d2c0c80145c Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Tue, 16 Jan 2024 15:17:16 -0500
Subject: [PATCH] add dunder methods to interface file

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

diff --git a/gtsam/gtsam.i b/gtsam/gtsam.i
index 834d5a147..9cead2a1c 100644
--- a/gtsam/gtsam.i
+++ b/gtsam/gtsam.i
@@ -39,6 +39,11 @@ class KeyList {
   void remove(size_t key);
 
   void serialize() const;
+
+  // Specual dunder methods for Python wrapping
+  __len__();
+  __contains__(size_t key);
+  __iter__();
 };
 
 // Actually a FastSet<Key>
@@ -64,6 +69,11 @@ class KeySet {
   bool count(size_t key) const;  // returns true if value exists
 
   void serialize() const;
+
+  // Specual dunder methods for Python wrapping
+  __len__();
+  __contains__(size_t key);
+  __iter__();
 };
 
 // Actually a vector<Key>, needed for Matlab
@@ -85,6 +95,11 @@ class KeyVector {
   void push_back(size_t key) const;
 
   void serialize() const;
+
+  // Specual dunder methods for Python wrapping
+  __len__();
+  __contains__(size_t key);
+  __iter__();
 };
 
 // Actually a FastMap<Key,int>