From 34dc163329d83b3bf42f2dbe65a9db0cd3be9e80 Mon Sep 17 00:00:00 2001 From: Casper da Costa-Luis Date: Sun, 18 Aug 2019 00:19:34 +0100 Subject: [PATCH] Tqdm auto leave=False for nested bars --- dvc/progress.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/dvc/progress.py b/dvc/progress.py index e3fd74200..04e075e3d 100644 --- a/dvc/progress.py +++ b/dvc/progress.py @@ -39,6 +39,7 @@ class Tqdm(tqdm): disable=None, bytes=False, # pylint: disable=W0622 desc_truncate=None, + leave=None, **kwargs ): """ @@ -63,6 +64,7 @@ class Tqdm(tqdm): super(Tqdm, self).__init__( iterable=iterable, disable=disable, **kwargs ) + self.leave = self.pos == 0 if leave is None else leave def update_desc(self, desc, n=1, truncate=True): """