Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "22ed39f57113cb8bed6baa464238b06550b5b0e6"
Unverified Commit ae743dcc authored by Da Zheng's avatar Da Zheng Committed by GitHub
Browse files

remove tqdm. (#1051)

parent dca0e376
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
apps/kg/dataloader/KGDataset.py
View file @ ae743dcc
...@@ -2,7 +2,6 @@ import os ...@@ -2,7 +2,6 @@ import os
def _download_and_extract(url, path, filename): def _download_and_extract(url, path, filename):
import shutil, zipfile import shutil, zipfile
from tqdm import tqdm
import requests import requests
fn = os.path.join(path, filename) fn = os.path.join(path, filename)
...@@ -19,7 +18,7 @@ def _download_and_extract(url, path, filename): ...@@ -19,7 +18,7 @@ def _download_and_extract(url, path, filename):
sz = f_remote.headers.get('content-length') sz = f_remote.headers.get('content-length')
assert f_remote.status_code == 200, 'fail to open {}'.format(url) assert f_remote.status_code == 200, 'fail to open {}'.format(url)
with open(fn, 'wb') as writer: with open(fn, 'wb') as writer:
for chunk in tqdm(f_remote.iter_content(chunk_size=1024*1024)): for chunk in f_remote.iter_content(chunk_size=1024*1024):
writer.write(chunk) writer.write(chunk)
print('Download finished. Unzipping the file...') print('Download finished. Unzipping the file...')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment