From efd8a070ff796260273ff257255f828be6a90799 Mon Sep 17 00:00:00 2001 From: Casper da Costa-Luis Date: Sun, 28 Jul 2019 15:39:35 +0100 Subject: [PATCH] tqdm remote.local --- dvc/remote/local/__init__.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/dvc/remote/local/__init__.py b/dvc/remote/local/__init__.py index 816cdaa2e..9de7c445b 100644 --- a/dvc/remote/local/__init__.py +++ b/dvc/remote/local/__init__.py @@ -35,7 +35,7 @@ from dvc.utils import ( ) from dvc.config import Config from dvc.exceptions import DvcException -from dvc.progress import progress +from dvc.progress import Tqdm from concurrent.futures import ThreadPoolExecutor from dvc.path_info import PathInfo @@ -255,7 +255,7 @@ class RemoteLOCAL(RemoteBASE): def cache_exists(self, checksums, jobs=None): return [ checksum - for checksum in progress(checksums) + for checksum in Tqdm(checksums, unit='md5') if not self.changed_cache_file(checksum) ] @@ -348,8 +348,8 @@ class RemoteLOCAL(RemoteBASE): cache = [] path_infos = [] names = [] - for md5, info in progress( - status_info.items(), name="Analysing status" + for md5, info in Tqdm( + status_info.items(), desc="Analysing status" ): if info["status"] == status: cache.append(self.checksum_to_path_info(md5)) @@ -542,7 +542,7 @@ class RemoteLOCAL(RemoteBASE): def _create_unpacked_dir(self, checksum, dir_info, unpacked_dir_info): self.makedirs(unpacked_dir_info) - for entry in progress(dir_info, name="Created unpacked dir"): + for entry in Tqdm(dir_info, desc="Created unpacked dir"): entry_cache_info = self.checksum_to_path_info( entry[self.PARAM_CHECKSUM] )