Remove sleep and use AsyncResult.wait instead of it

ieasybooks · Aug 25, 2023 · aa6214c · aa6214c
1 parent 15a19c7
commit aa6214c
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/tafrigh/recognizers/wit_recognizer.py b/tafrigh/recognizers/wit_recognizer.py
@@ -86,10 +86,11 @@ def recognize(
                 transcriptions = []
 
                 with tqdm(total=len(segments), disable=self.verbose is not False) as pbar:
-                    while async_results:
-                        if async_results[0].ready():
-                            transcriptions.append(async_results.pop(0).get())
-                            pbar.update(1)
+                    for async_result in async_results:
+                        async_result.wait()
+                        pbar.update(1)
+
+                        transcriptions.append(async_result.get())
 
                         yield {
                             'progress': round(len(transcriptions) / len(segments) * 100, 2),
@@ -98,8 +99,6 @@ def recognize(
                             else None,
                         }
 
-                        time.sleep(0.1)
-
         shutil.rmtree(temp_directory)
 
         return transcriptions