未验证 提交 5a0103b2 编写于 作者: P PiaoYang 提交者: GitHub

[BUG] Fix progress bar unit. (#3177)

上级 6ffd3e4f
...@@ -133,10 +133,10 @@ def _get_download(url, fullname): ...@@ -133,10 +133,10 @@ def _get_download(url, fullname):
total_size = req.headers.get('content-length') total_size = req.headers.get('content-length')
with open(tmp_fullname, 'wb') as f: with open(tmp_fullname, 'wb') as f:
if total_size: if total_size:
with tqdm(total=(int(total_size) + 1023) // 1024) as pbar: with tqdm(total=(int(total_size)), unit='B', unit_scale=True) as pbar:
for chunk in req.iter_content(chunk_size=1024): for chunk in req.iter_content(chunk_size=1024):
f.write(chunk) f.write(chunk)
pbar.update(1) pbar.update(len(chunk))
else: else:
for chunk in req.iter_content(chunk_size=1024): for chunk in req.iter_content(chunk_size=1024):
if chunk: if chunk:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册