From c1ff4c29b92cae9d66fca9294f3b248286a34ae4 Mon Sep 17 00:00:00 2001 From: megemini Date: Tue, 24 Dec 2024 15:27:56 +0800 Subject: [PATCH] [Update] assert empyt folder --- .../vector/exps/ge2e/speaker_verification_dataset.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/paddlespeech/vector/exps/ge2e/speaker_verification_dataset.py b/paddlespeech/vector/exps/ge2e/speaker_verification_dataset.py index 873ebfeef..ae2c9427f 100644 --- a/paddlespeech/vector/exps/ge2e/speaker_verification_dataset.py +++ b/paddlespeech/vector/exps/ge2e/speaker_verification_dataset.py @@ -39,10 +39,12 @@ class MultiSpeakerMelDataset(Dataset): def __init__(self, dataset_root: Path): self.root = Path(dataset_root).expanduser() - speaker_dirs = [ - f for f in self.root.glob("*") - if f.is_dir() and list(f.glob("*.npy")) - ] + speaker_dirs = [] + for f in self.root.glob("*"): + if f.is_dir(): + assert list(f.glob( + "*.npy")), "This folder NOT includes any npy data file." + speaker_dirs.append(f) speaker_utterances = { speaker_dir: list(speaker_dir.glob("*.npy"))