mawei/sunnypilot
1
0
Fork 0
mirror of https://github.com/sunnypilot/sunnypilot.git synced 2026-02-27 18:03:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
old-commit-hash: 9e648e428d
This commit is contained in:
Hoang Bui
2024-08-08 10:51:14 -04:00
committed by GitHub
parent a7db075f08
commit 6a2b4dedfc
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tools/lib/logreader.py
View File

@@ -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