Stone_Red/dvc
1
0
Fork 0
mirror of https://github.com/Stone-Red-Code/dvc.git synced 2026-11-01 23:08:43 +01:00
Code Issues Packages Projects Releases Wiki Activity

better auto-closing TqdmThreadPoolExecutor

Browse Source
This commit is contained in:
Casper da Costa-Luis
2019-08-13 23:46:20 +01:00
parent a1f5c03683
commit d9b4ecfe7b
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dvc/progress.py
+8 -4
View File
@@ -13,15 +13,19 @@ class TqdmThreadPoolExecutor(ThreadPoolExecutor):
def __enter__(self):
"""
Creates a blank initial dummy progress bar so that workers are forced
to create "nested" bars.
Creates a blank initial dummy progress bar if needed so that workers are
forced to create "nested" bars.
"""
self.blank_bar = Tqdm(bar_format="Multi-Threaded:", leave=False)
blank_bar = Tqdm(bar_format="Multi-Threaded:", leave=False)
if blank_bar.pos > 0:
# already nested - don't need a placeholder bar
blank_bar.close()
self.bar = blank_bar
return super(TqdmThreadPoolExecutor, self).__enter__()
def __exit__(self, *a, **k):
super(TqdmThreadPoolExecutor, self).__exit__(*a, **k)
self.blank_bar.close()
self.bar.close()
class Tqdm(tqdm):