Tqdm fix

sweepai · Mar 14, 2024 · 3ac85de · 3ac85de
1 parent 3ec3b32
commit 3ac85de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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