tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-30 09:26:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

improve progress bar

Browse Source
This commit is contained in:
geohot
2023-01-30 14:50:28 -08:00
parent 4685c9c095
commit c6f570a2e6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extra/utils.py
+1 -1
View File
@@ -18,7 +18,7 @@ def fetch(url):
r = requests.get(url, stream=True)
assert r.status_code == 200
progress_bar = tqdm(total=int(r.headers.get('content-length', 0)), unit='B', unit_scale=True, desc=url)
dat = b''.join(x for x in r.iter_content(chunk_size=16384) if progress_bar.update(len(x)) or True)
dat = b''.join((x,progress_bar.update(len(x)))[0] for x in r.iter_content(chunk_size=16384))
with open(fp+".tmp", "wb") as f:
f.write(dat)
os.rename(fp+".tmp", fp)