From 50011f43cd5c560d5730d9af91ac49f0b6690a6f Mon Sep 17 00:00:00 2001 From: atila Date: Fri, 8 Nov 2024 18:37:14 -0800 Subject: [PATCH] Fixes #15 --- whisperkit/evaluate/evaluate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/whisperkit/evaluate/evaluate.py b/whisperkit/evaluate/evaluate.py index f3f3ea3..576f168 100644 --- 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) ))