0.4.4: fix flatten non yielding on single input iterator with quite f…

…ast iteration
ebonnal · Oct 17, 2023 · 0f9c02f · 0f9c02f
1 parent 8a88c2e
commit 0f9c02f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/kioss/_exec.py b/kioss/_exec.py
@@ -64,6 +64,8 @@ def __iter__(self):
                         not iterator_exhausted
                         and executor._work_queue.qsize()
                         < ThreadedMappingIterable._MAX_QUEUE_SIZE
+                        and n_iterated_elems - n_yields
+                        < ThreadedMappingIterable._MAX_QUEUE_SIZE
                     ):
                         try:
                             elem = next(self.iterator)

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='kioss',
-    version='0.4.3',
+    version='0.4.4',
     packages=['kioss'],
     url='http://github.com/bonnal-enzo/kioss',
     license='Apache 2.',