disable tqdm

pull/34936/head
Shane Smiskol 2 months ago
parent 6798395405
commit 593f8c8864
  1. 2
      selfdrive/debug/max_lat_accel.py
  2. 4
      tools/lib/logreader.py

@ -99,7 +99,7 @@ if __name__ == '__main__':
print('WARNING: Treating route as qlog!') print('WARNING: Treating route as qlog!')
print('Finding events...') 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()
print(f'Found {len(events)} events') print(f'Found {len(events)} events')

@ -307,11 +307,11 @@ class LogReader:
def _run_on_segment(self, func, i): def _run_on_segment(self, func, i):
return func(self._get_lr(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: with multiprocessing.Pool(num_processes) as pool:
ret = [] ret = []
num_segs = len(self.logreader_identifiers) 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) ret.extend(p)
return ret return ret

Loading…
Cancel
Save