From 3ac85de05271ffff0d7e26f90fd642e65e82e530 Mon Sep 17 00:00:00 2001 From: Kevin Lu Date: Thu, 14 Mar 2024 07:04:58 +0000 Subject: [PATCH] Tqdm fix --- sweepai/core/repo_parsing_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sweepai/core/repo_parsing_utils.py b/sweepai/core/repo_parsing_utils.py index 99c28be674..466b289f10 100644 --- a/sweepai/core/repo_parsing_utils.py +++ b/sweepai/core/repo_parsing_utils.py @@ -120,6 +120,6 @@ def is_dir_too_big(file_name): logger.info("Done reading files") all_chunks = [] with multiprocessing.Pool(processes=multiprocessing.cpu_count() // 4) as pool: - for chunks in tqdm(pool.imap(file_path_to_chunks, file_list)): + for chunks in tqdm(pool.imap(file_path_to_chunks, file_list), total=len(file_list)): all_chunks.extend(chunks) return all_chunks, file_list