add saving progress for files in dirs

iterative · Jan 26, 2020 · a66cc0e · a66cc0e
1 parent 25557c5
commit a66cc0e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dvc/remote/base.py b/dvc/remote/base.py
@@ -493,7 +493,9 @@ def _save_dir(self, path_info, checksum):
         cache_info = self.checksum_to_path_info(checksum)
         dir_info = self.get_dir_cache(checksum)
 
-        for entry in dir_info:
+        for entry in Tqdm(
+            dir_info, desc="Saving " + path_info.name, unit="file"
+        ):
             entry_info = path_info / entry[self.PARAM_RELPATH]
             entry_checksum = entry[self.PARAM_CHECKSUM]
             self._save_file(entry_info, entry_checksum, save_link=False)