linter fix

This commit is contained in:
Eren Gölge 2021-03-08 05:29:43 +01:00
parent 462fb5e6ab
commit e15734c3fc
2 changed files with 2 additions and 0 deletions

View File

@ -53,6 +53,7 @@ class MyDataset(Dataset):
use_noise_augment (bool): enable adding random noise to wav for augmentation.
verbose (bool): print diagnostic information.
"""
super(MyDataset, self).__init__()
self.batch_group_size = batch_group_size
self.items = meta_data
self.outputs_per_step = outputs_per_step

View File

@ -21,6 +21,7 @@ class WaveRNNDataset(Dataset):
verbose=False,
):
super(WaveRNNDataset, self).__init__()
self.ap = ap
self.compute_feat = not isinstance(items[0], (tuple, list))
self.item_list = items