From 866b3cd0ae89d6155e78a69f86e871300f63c52d Mon Sep 17 00:00:00 2001 From: Casper da Costa-Luis Date: Wed, 27 Nov 2019 23:24:09 +0000 Subject: [PATCH] add progress for save/commit/dump stages --- dvc/repo/add.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/dvc/repo/add.py b/dvc/repo/add.py index b3d24eefb..bd615860a 100644 --- a/dvc/repo/add.py +++ b/dvc/repo/add.py @@ -50,7 +50,7 @@ def add(repo, targets, recursive=False, no_commit=False, fname=None): repo.check_modified_graph(stages) - for stage in stages: + for stage in Tqdm(stages, desc="Processing", unit="file"): stage.save() if not no_commit: @@ -88,7 +88,10 @@ def _create_stages(repo, targets, fname, pbar=None): stages = [] for out in Tqdm( - targets, desc="Creating stages", disable=len(targets) < LARGE_DIR_SIZE + targets, + desc="Creating stages", + disable=len(targets) < LARGE_DIR_SIZE, + unit="file", ): stage = Stage.create(repo, outs=[out], add=True, fname=fname)