From 24e91b9e1b363a75a98afe48c3ec94b5874226e0 Mon Sep 17 00:00:00 2001
From: Yuval Adam <_@yuv.al>
Date: Fri, 21 Mar 2025 22:33:33 +0100
Subject: Cleanup formatting

---
 rivulet/pipeline.py | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/rivulet/pipeline.py b/rivulet/pipeline.py
index 226a645..6efd821 100644
--- a/rivulet/pipeline.py
+++ b/rivulet/pipeline.py
@@ -1,8 +1,9 @@
-from typing import AsyncGenerator, TypeVar, Callable, List, Generic, Any
+from typing import AsyncGenerator, TypeVar, Callable, List, Any
+
+T = TypeVar("T")
+U = TypeVar("U")
+V = TypeVar("V")
 
-T = TypeVar('T')
-U = TypeVar('U')
-V = TypeVar('V')
 
 class Pipeline:
     """
@@ -14,9 +15,14 @@ class Pipeline:
     def __init__(self, source: AsyncGenerator[Any, None]):
         """Initialize the pipeline with a source async generator."""
         self.source = source
-        self.steps: List[Callable[[AsyncGenerator[Any, None]], AsyncGenerator[Any, None]]] = []
+        self.steps: List[
+            Callable[[AsyncGenerator[Any, None]], AsyncGenerator[Any, None]]
+        ] = []
 
-    def add_step(self, transform: Callable[[AsyncGenerator[Any, None]], AsyncGenerator[Any, None]]):
+    def add_step(
+        self,
+        transform: Callable[[AsyncGenerator[Any, None]], AsyncGenerator[Any, None]],
+    ):
         """
         Add a transformation step to the pipeline.
 
@@ -60,4 +66,4 @@ class Pipeline:
         results = []
         async for item in self:
             results.append(item)
-        return results
\ No newline at end of file
+        return results
-- 
cgit v1.3.1