Added __len__ to VariablesDict

AmitSolomonPrinceton · AmitSolomonPrinceton · commit 1c20805189e8 · 2025-03-10T12:04:07.000-04:00
diff --git a/cvxtorch/variables_dict/variables_dict.py b/cvxtorch/variables_dict/variables_dict.py
@@ -6,7 +6,10 @@ def __init__(self, provided_vars_list: list=[]):
         self.vars_dict = dict()
         for var in provided_vars_list:
             self.add_var(var)
-            
+    
+    def __len__(self) -> int:
+        return len(self.vars_dict)
+
     def add_var(self, var):
         """
         var is expected to be either a cp.Variable or a cp.Parameter, but is not enforced.
