ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-01 09:36:49 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix download progress bar based on sdk.storage.log.report_download_chunk_size_mb configuration

Browse Source
This commit is contained in:
allegroai 2021-11-29 12:46:40 +02:00
parent 144832d001
commit 73cfbcb947
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
clearml/storage/helper.py
View File

@ -685,8 +685,7 @@ class StorageHelper(object):
# if driver supports download with callback, use it (it might be faster) # if driver supports download with callback, use it (it might be faster)
if hasattr(self._driver, 'download_object'): if hasattr(self._driver, 'download_object'):
# callback # callback
cb = DownloadProgressReport(total_size_mb, verbose, cb = DownloadProgressReport(total_size_mb, verbose, remote_path, self._log)
remote_path, self._log, chunk_size_mb)
self._driver.download_object(obj, temp_local_path, callback=cb) self._driver.download_object(obj, temp_local_path, callback=cb)
download_reported = bool(cb.last_reported) download_reported = bool(cb.last_reported)
dl_total_mb = cb.current_status_mb dl_total_mb = cb.current_status_mb