From 7318a6d120a3fa889fea1e6ee0a794e094027158 Mon Sep 17 00:00:00 2001 From: Casper da Costa-Luis Date: Mon, 2 Dec 2019 17:35:29 +0000 Subject: [PATCH] safer download notification --- dvc/remote/gdrive.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/dvc/remote/gdrive.py b/dvc/remote/gdrive.py index f29b436c1..e2cd6f57b 100644 --- a/dvc/remote/gdrive.py +++ b/dvc/remote/gdrive.py @@ -115,10 +115,11 @@ class RemoteGDrive(RemoteBASE): self, file_id, to_file, progress_name, no_progress_bar ): gdrive_file = self.drive.CreateFile({"id": file_id}) + bar_format = "Donwloading {desc}... " + Tqdm.format_sizeof( + int(gdrive_file["fileSize"]), "B", 1024 + ) with Tqdm( - desc=progress_name, - total=int(gdrive_file["fileSize"]), - disable=no_progress_bar, + bar_format=bar_format, desc=progress_name, disable=no_progress_bar ): gdrive_file.GetContentFile(to_file)