From 120f087f1c3bb4879e4be2069cf5617cced40fe6 Mon Sep 17 00:00:00 2001
From: Vera Sativa <hola@verasativa.com>
Date: Fri, 6 Dec 2019 10:04:05 -0300
Subject: [PATCH] fix: tqdm reading futures

---
 dvc/remote/base.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/dvc/remote/base.py b/dvc/remote/base.py
index d6c688bf1..c9365dd1c 100644
--- a/dvc/remote/base.py
+++ b/dvc/remote/base.py
@@ -573,14 +573,16 @@ class RemoteBASE(object):
                 file_mode=file_mode,
                 dir_mode=dir_mode,
             )
+            futures = executor.map(
+                download_files, file_from_info, file_to_infos
+            )
             with Tqdm(
-                file_from_info,
+                futures,
                 total=len(file_from_info),
                 desc="Downloading directory",
-            ) as file_from_info:
-                return sum(
-                    executor.map(download_files, file_from_info, file_to_infos)
-                )
+                unit="Files",
+            ) as futures:
+                return sum(futures)
 
     def _download_file(
         self, from_info, to_info, name, no_progress_bar, file_mode, dir_mode