[Update] assert empyt folder

pull/3948/head
megemini 9 months ago
parent cb05ca2979
commit c1ff4c29b9

@ -39,10 +39,12 @@ class MultiSpeakerMelDataset(Dataset):
def __init__(self, dataset_root: Path): def __init__(self, dataset_root: Path):
self.root = Path(dataset_root).expanduser() self.root = Path(dataset_root).expanduser()
speaker_dirs = [ speaker_dirs = []
f for f in self.root.glob("*") for f in self.root.glob("*"):
if f.is_dir() and list(f.glob("*.npy")) if f.is_dir():
] assert list(f.glob(
"*.npy")), "This folder NOT includes any npy data file."
speaker_dirs.append(f)
speaker_utterances = { speaker_utterances = {
speaker_dir: list(speaker_dir.glob("*.npy")) speaker_dir: list(speaker_dir.glob("*.npy"))

Loading…
Cancel
Save