Merge pull request #336 from Miuzarte/4.1-Stable

why not rich.progress
This commit is contained in:
謬紗特 2023-07-23 09:16:43 +08:00 committed by GitHub
commit b0e7e83880
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -7,7 +7,7 @@ from multiprocessing import cpu_count
import librosa
import numpy as np
from scipy.io import wavfile
from tqdm import tqdm
from rich.progress import track
def load_wav(wav_path):
@ -81,7 +81,7 @@ def process_all_speakers():
if os.path.isdir(spk_dir):
print(spk_dir)
futures = [executor.submit(process, (spk_dir, i, args)) for i in os.listdir(spk_dir) if i.endswith("wav")]
for _ in tqdm(concurrent.futures.as_completed(futures), total=len(futures)):
for _ in track(concurrent.futures.as_completed(futures), total=len(futures), description="resampling:"):
pass