From 593f8c88649e2fb7fe03503855e303f48b058f1e Mon Sep 17 00:00:00 2001 From: Shane Smiskol Date: Tue, 25 Mar 2025 13:23:55 -0700 Subject: [PATCH] disable tqdm --- selfdrive/debug/max_lat_accel.py | 2 +- tools/lib/logreader.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/selfdrive/debug/max_lat_accel.py b/selfdrive/debug/max_lat_accel.py index 9275ca5aa6..dc44e8ac40 100755 --- a/selfdrive/debug/max_lat_accel.py +++ b/selfdrive/debug/max_lat_accel.py @@ -99,7 +99,7 @@ if __name__ == '__main__': print('WARNING: Treating route as qlog!') print('Finding events...') - events += lr.run_across_segments(8, partial(find_events, extrapolate=args.extrapolate, qlog=qlog)) + events += lr.run_across_segments(8, partial(find_events, extrapolate=args.extrapolate, qlog=qlog), disable_tqdm=True) print() print(f'Found {len(events)} events') diff --git a/tools/lib/logreader.py b/tools/lib/logreader.py index 290171d363..2a0015e6b0 100755 --- a/tools/lib/logreader.py +++ b/tools/lib/logreader.py @@ -307,11 +307,11 @@ class LogReader: def _run_on_segment(self, func, i): return func(self._get_lr(i)) - def run_across_segments(self, num_processes, func, desc=None): + def run_across_segments(self, num_processes, func, disable_tqdm=False, desc=None): with multiprocessing.Pool(num_processes) as pool: ret = [] num_segs = len(self.logreader_identifiers) - for p in tqdm.tqdm(pool.imap(partial(self._run_on_segment, func), range(num_segs)), total=num_segs, desc=desc): + for p in tqdm.tqdm(pool.imap(partial(self._run_on_segment, func), range(num_segs)), total=num_segs, disable=disable_tqdm, desc=desc): ret.extend(p) return ret