Fixes #15

argmaxinc · Nov 9, 2024 · 50011f4 · 50011f4
1 parent 8378fb3
commit 50011f4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/whisperkit/evaluate/evaluate.py b/whisperkit/evaluate/evaluate.py
@@ -54,7 +54,7 @@ def evaluate(whisper_pipeline: Union[pipelines.WhisperPipeline, pipelines.Whispe
 
         logger.info(f"Launching {num_proc} processes to run {whisper_pipeline.__class__.__name__}")
         with Pool(num_proc) as pool:
-            results = list(tqdm.tqdm(pool.map(
+            results = list(tqdm.tqdm(pool.imap(
                 partial(evaluate_sample, whisper_pipeline=whisper_pipeline), dataset),
                 total=len(dataset)
             ))