Add __iter__ to Aggify

Aggify · Nov 1, 2023 · fab2ea3 · fab2ea3
1 parent 18a1c0a
commit fab2ea3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/aggify/aggify.py b/aggify/aggify.py
@@ -61,6 +61,10 @@ def __init__(self, base_model: Type[Document]):
         self.stop = None
         self.q = None
 
+    def __iter__(self):
+        # Return a generator or iterator for the data you want to represent as a list
+        return iter(self.pipelines)
+
     @last_out_stage_check
     def project(self, **kwargs: QueryParams) -> "Aggify":
         """