concurrent: simplify results loop

2026-05-04 20:54:18 -04:00 · 2021-01-01 00:44:46 -08:00
parent bd0f005861
commit df6e106203
1 changed files with 5 additions and 9 deletions
--- a/src/ocrmypdf/_concurrent.py
+++ b/src/ocrmypdf/_concurrent.py
@@ -109,15 +109,11 @@ def exec_progress_pool(
        )
        try:
            results = pool.imap_unordered(task, task_arguments)
-            while True:
-                try:
-                    result = results.next()
-                    if task_finished:
-                        task_finished(result, pbar)
-                    else:
-                        pbar.update()
-                except StopIteration:
-                    break
+            for result in results:
+                if task_finished:
+                    task_finished(result, pbar)
+                else:
+                    pbar.update()
        except KeyboardInterrupt:
            # Terminate pool so we exit instantly
            pool.terminate()