linter fixes

pull/367/head
Eren Gölge 2021-03-08 11:22:59 +01:00
parent 988ae7a510
commit 594d8d8f09
2 changed files with 2 additions and 0 deletions

View File

@ -17,6 +17,7 @@ class MyDataset(Dataset):
seq_len (int): voice segment length in seconds.
verbose (bool): print diagnostic information.
"""
super().__init__()
self.items = meta_data
self.sample_rate = ap.sample_rate
self.voice_len = voice_len

View File

@ -26,6 +26,7 @@ class WaveGradDataset(Dataset):
use_cache=False,
verbose=False):
super().__init__()
self.ap = ap
self.item_list = items
self.seq_len = seq_len if return_segments else None