From caae46cfbaffbd262ad44c7aaae213b1885b907d Mon Sep 17 00:00:00 2001 From: qazal <77887910+Qazalin@users.noreply.github.com> Date: Fri, 10 Oct 2025 10:20:55 +0300 Subject: [PATCH] fix process replay progress update (#12587) --- test/external/process_replay/process_replay.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/external/process_replay/process_replay.py b/test/external/process_replay/process_replay.py index 55e06eef2d..3ffe5f70b7 100755 --- a/test/external/process_replay/process_replay.py +++ b/test/external/process_replay/process_replay.py @@ -114,7 +114,7 @@ def _pmap(fxns:dict[str, Callable]) -> None: with multiprocessing.get_context("spawn").Pool(multiprocessing.cpu_count()) as pool: bar = tqdm(total=row_count) - for _ in pool.imap_unordered(functools.partial(diff, fxns=fxns), range(0, row_count, PAGE_SIZE)): bar.update(PAGE_SIZE) + for _ in pool.imap_unordered(functools.partial(diff, fxns=fxns), range(0, row_count, s:=min(PAGE_SIZE, row_count))): bar.update(s) pool.close() pool.join() pool.terminate()