From df6e1062033003ca3dcf5463b95d04b74d00481b Mon Sep 17 00:00:00 2001
From: "James R. Barlow" <james@purplerock.ca>
Date: Fri, 1 Jan 2021 00:44:46 -0800
Subject: [PATCH] concurrent: simplify results loop

---
 src/ocrmypdf/_concurrent.py | 14 +++++---------
 1 file changed, 5 insertions(+), 9 deletions(-)

diff --git a/src/ocrmypdf/_concurrent.py b/src/ocrmypdf/_concurrent.py
index 70ee03dd..177ffe0a 100644
--- 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()