From 7171712fa998820d0fa3b9bb19ce7a9bf378011f Mon Sep 17 00:00:00 2001 From: Casper da Costa-Luis Date: Tue, 27 Aug 2019 13:41:37 +0100 Subject: [PATCH] delay bar_format --- dvc/progress.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/dvc/progress.py b/dvc/progress.py index ffef29255..c0a67b323 100644 --- a/dvc/progress.py +++ b/dvc/progress.py @@ -81,19 +81,21 @@ class Tqdm(tqdm): disable = logger.getEffectiveLevel() > level if leave is None: leave = logger.getEffectiveLevel() <= level_leave - if bar_format is None: - if kwargs.get("total", hasattr(iterable, "__len__")): - bar_format = self.BAR_FMT_DEFAULT - else: - bar_format = self.BAR_FMT_NOTOTAL super(Tqdm, self).__init__( iterable=iterable, disable=disable, leave=leave, desc=desc, - bar_format=bar_format, + bar_format="", **kwargs ) + if bar_format is None: + self.bar_format = ( + self.BAR_FMT_DEFAULT if len(self) else self.BAR_FMT_NOTOTAL + ) + else: + self.bar_format = bar_format + self.refresh() def update_desc(self, desc, n=1, truncate=True): """