diff --git a/dvc/tree/ssh/connection.py b/dvc/tree/ssh/connection.py index c313eafe02..e3b236f5bc 100644 --- a/dvc/tree/ssh/connection.py +++ b/dvc/tree/ssh/connection.py @@ -189,6 +189,7 @@ def download(self, src, dest, no_progress_bar=False, progress_title=None): desc=progress_title or os.path.basename(src), disable=no_progress_bar, bytes=True, + total=self.getsize(src), ) as pbar: self.sftp.get(src, dest, callback=pbar.update_to) @@ -221,7 +222,10 @@ def upload(self, src, dest, no_progress_bar=False, progress_title=None): progress_title = posixpath.basename(dest) with Tqdm( - desc=progress_title, disable=no_progress_bar, bytes=True + desc=progress_title, + disable=no_progress_bar, + bytes=True, + total=os.path.getsize(src), ) as pbar: self.sftp.put(src, tmp_file, callback=pbar.update_to)