Change location from which to import tqdm (speechbrain#2921)

rogiervd · Rogier van Dalen · pplantinga · commit fbed80590726 · 2025-06-02T14:12:30.000-04:00
Co-authored-by: Rogier van Dalen &lt;r.vandalen@samsung.com&gt;
diff --git a/recipes/AMI/Diarization/experiment.py b/recipes/AMI/Diarization/experiment.py
@@ -28,7 +28,7 @@
 import numpy as np
 import torch
 from hyperpyyaml import load_hyperpyyaml
-from tqdm.contrib import tqdm
+from tqdm import tqdm
 
 import speechbrain as sb
 from speechbrain.dataio.dataio import read_audio, read_audio_multichannel
diff --git a/recipes/IEMOCAP/emotion_recognition/train.py b/recipes/IEMOCAP/emotion_recognition/train.py
@@ -19,7 +19,7 @@
 import torch
 from hyperpyyaml import load_hyperpyyaml
 from torch.utils.data import DataLoader
-from tqdm.contrib import tqdm
+from tqdm import tqdm
 
 import speechbrain as sb
 
diff --git a/recipes/VoxCeleb/SpeakerRec/speaker_verification_cosine.py b/recipes/VoxCeleb/SpeakerRec/speaker_verification_cosine.py
@@ -18,7 +18,7 @@
 import torch
 import torchaudio
 from hyperpyyaml import load_hyperpyyaml
-from tqdm.contrib import tqdm
+from tqdm import tqdm
 
 import speechbrain as sb
 from speechbrain.utils.data_utils import download_file
diff --git a/recipes/VoxCeleb/SpeakerRec/speaker_verification_plda.py b/recipes/VoxCeleb/SpeakerRec/speaker_verification_plda.py
@@ -19,7 +19,7 @@
 import torch
 import torchaudio
 from hyperpyyaml import load_hyperpyyaml
-from tqdm.contrib import tqdm
+from tqdm import tqdm
 
 import speechbrain as sb
 from speechbrain.processing.PLDA_LDA import (
diff --git a/recipes/VoxCeleb/voxceleb_prepare.py b/recipes/VoxCeleb/voxceleb_prepare.py
@@ -14,7 +14,7 @@
 import numpy as np
 import torch
 import torchaudio
-from tqdm.contrib import tqdm
+from tqdm import tqdm
 
 from speechbrain.dataio.dataio import load_pkl, save_pkl
 from speechbrain.utils.logger import get_logger
diff --git a/speechbrain/core.py b/speechbrain/core.py
@@ -33,7 +33,7 @@
 from torch.nn import SyncBatchNorm
 from torch.nn.parallel import DistributedDataParallel as DDP
 from torch.utils.data import DataLoader, DistributedSampler, IterableDataset
-from tqdm.contrib import tqdm
+from tqdm import tqdm
 
 import speechbrain as sb
 from speechbrain.dataio.dataloader import LoopedLoader, SaveableDataLoader
diff --git a/speechbrain/utils/kmeans.py b/speechbrain/utils/kmeans.py
@@ -7,7 +7,7 @@
 
 import os
 
-from tqdm.contrib import tqdm
+from tqdm import tqdm
 
 from speechbrain.utils.logger import get_logger
 
diff --git a/tests/templates/fetching_ddp_dynbatch_finetuning/finetune.py b/tests/templates/fetching_ddp_dynbatch_finetuning/finetune.py
@@ -15,7 +15,7 @@
 from ASR_template_train import ASR, dataio_prepare
 from hyperpyyaml import load_hyperpyyaml
 from torch.utils.data import DataLoader
-from tqdm.contrib import tqdm
+from tqdm import tqdm
 
 import speechbrain as sb
 from speechbrain.dataio.dataio import read_audio  # read_audio_multichannel,
diff --git a/tests/templates/fetching_ddp_dynbatch_finetuning/finetune_fetch_once.py b/tests/templates/fetching_ddp_dynbatch_finetuning/finetune_fetch_once.py
@@ -15,7 +15,7 @@
 from ASR_template_train import ASR, dataio_prepare
 from hyperpyyaml import load_hyperpyyaml
 from torch.utils.data import DataLoader
-from tqdm.contrib import tqdm
+from tqdm import tqdm
 
 import speechbrain as sb
 from speechbrain.dataio.dataio import read_audio  # read_audio_multichannel,
