tools/logreader: add desc arg to run_across_segments() (#33226)

pull/33236/head
Hoang Bui 12 months ago committed by GitHub
parent 9880b1393c
commit 9e648e428d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 4
      tools/lib/logreader.py

@ -293,11 +293,11 @@ are uploaded or auto fallback to qlogs with '/a' selector at the end of the rout
def _run_on_segment(self, func, i):
return func(self._get_lr(i))
def run_across_segments(self, num_processes, func):
def run_across_segments(self, num_processes, func, 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):
for p in tqdm.tqdm(pool.imap(partial(self._run_on_segment, func), range(num_segs)), total=num_segs, desc=desc):
ret.extend(p)
return ret

Loading…
Cancel
Save