From fb2543e81ee85658835ea820386f1009d76cd7fa Mon Sep 17 00:00:00 2001 From: shahules786 Date: Thu, 27 Oct 2022 16:18:31 +0530 Subject: [PATCH] fix typo --- enhancer/data/dataset.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py index dac2c50..8851ea2 100644 --- a/enhancer/data/dataset.py +++ b/enhancer/data/dataset.py @@ -124,10 +124,7 @@ class TaskDataset(pl.LightningDataModule): valid_sec_now = 0.0 valid_indices = [] all_speakers = np.unique( - [ - (Path(file["clean"]).name.split("_")[0], file["duration"]) - for file in data - ] + [Path(file["clean"]).name.split("_")[0] for file in data] ) possible_indices = list(range(0, len(all_speakers))) rng = create_unique_rng(len(all_speakers))