no unit scaling for small totals

iterative · Jan 26, 2020 · c6c8b80 · c6c8b80
1 parent 5d10606
commit c6c8b80
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/dvc/progress.py b/dvc/progress.py
@@ -47,6 +47,7 @@ def __init__(
         bar_format=None,
         bytes=False,  # pylint: disable=W0622
         file=None,
+        total=None,
         **kwargs
     ):
         """
@@ -61,9 +62,10 @@ def __init__(
         kwargs  : anything accepted by `tqdm.tqdm()`
         """
         kwargs = kwargs.copy()
-        kwargs.setdefault("unit_scale", True)
         if bytes:
             kwargs = merge(self.BYTES_DEFAULTS, kwargs)
+        else:
+            kwargs.setdefault("unit_scale", total > 999 if total else True)
         if file is None:
             file = sys.stderr
         self.desc_persist = desc
@@ -84,6 +86,7 @@ def __init__(
             desc=desc,
             bar_format="!",
             lock_args=(False,),
+            total=total,
             **kwargs
         )
         if bar_format is None:

diff --git a/dvc/repo/add.py b/dvc/repo/add.py
@@ -54,8 +54,6 @@ def add(repo, targets, recursive=False, no_commit=False, fname=None):
                 stage.dump()
 
             stages_list += stages
-        # remove filled bar bit of progress, leaving stats
-        pbar.bar_format = pbar.BAR_FMT_DEFAULT.replace("|{bar:10}|", " ")
 
     return stages_list