From 9b56aaa7a1aefc2fd07e3bf4baedf9206d9dd7ad Mon Sep 17 00:00:00 2001
From: Yuval Adam <_@yuv.al>
Date: Fri, 21 Mar 2025 22:16:47 +0100
Subject: Solidify Batch API

---
 example.py | 43 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 example.py

(limited to 'example.py')

diff --git a/example.py b/example.py
new file mode 100644
index 0000000..08fd8b0
--- /dev/null
+++ b/example.py
@@ -0,0 +1,43 @@
+import asyncio
+
+from rivulet import Pipeline, Batch
+
+
+async def source():
+    for i in range(22):
+        yield i
+
+
+async def double(gen):
+    async for value in gen:
+        yield value * 2
+
+
+async def dump(gen):
+    async for x in gen:
+        print(x)
+        yield x
+
+
+async def sum(batches):
+    async for batch in batches:
+        res = 0
+        for x in batch:
+            res += x
+        yield res
+
+
+async def main():
+    pipe = Pipeline(source())
+    pipe.add_step(double)
+
+    batch = Batch(N=5, timeout=0.1)
+    pipe.add_step(batch)
+    pipe.add_step(dump)
+    pipe.add_step(sum)
+
+    res = await pipe.collect()
+    print(res)
+
+
+asyncio.run(main())
-- 
cgit v1.3.1