Getting started on interruptible transport pipeline runner

2024-03-04 07:51:22 -05:00
parent 763a50f8ec
commit 18e7626b9f
5 changed files with 47 additions and 9 deletions
--- a/src/dailyai/tests/test_aggregators.py
+++ b/src/dailyai/tests/test_aggregators.py
@@ -87,9 +87,14 @@ class TestDailyFrameAggregators(unittest.IsolatedAsyncioTestCase):
        source = asyncio.Queue()
        sink = asyncio.Queue()
        pipeline = Pipeline(
+            [
+                ParallelPipeline(
+                    [[pipe1_annotation], [sentence_aggregator, pipe2_annotation]]
+                ),
+                add_dots,
+            ],
            source,
            sink,
-            [ParallelPipeline([[pipe1_annotation], [sentence_aggregator, pipe2_annotation]]), add_dots],
        )

        frames = [