diff --git a/datasets/binar_masks.py b/datasets/binar_masks.py index f587ba6..38477db 100644 --- a/datasets/binar_masks.py +++ b/datasets/binar_masks.py @@ -58,7 +58,11 @@ class BinaryMasksDataset(Dataset): # Delete File if one exists. if not self.use_preprocessed: for key in labeldict.keys(): - (self._mel_folder / (key.replace('.wav', '') + self.container_ext)).unlink(missing_ok=True) + try: + (self._mel_folder / (key.replace('.wav', '') + self.container_ext)).unlink() + except FileNotFoundError: + pass + return labeldict def __len__(self):