format with black and pylint 2.7.3

This commit is contained in:
Eren Gölge
2021-04-09 00:38:08 +02:00
parent 5de7eb708b
commit 0e79fa86ad
127 changed files with 5511 additions and 5491 deletions
+1 -1
View File
@@ -17,7 +17,7 @@ conf = load_config(os.path.join(get_tests_input_path(), 'test_config.json'))
# pylint: disable=protected-access
class TestAudio(unittest.TestCase):
def __init__(self, *args, **kwargs):
super(TestAudio, self).__init__(*args, **kwargs)
super().__init__(*args, **kwargs)
self.ap = AudioProcessor(**conf.audio)
def test_audio_synthesis(self):
+1 -1
View File
@@ -28,7 +28,7 @@ print(" > Dynamic data loader test: {}".format(DATA_EXIST))
class TestTTSDataset(unittest.TestCase):
def __init__(self, *args, **kwargs):
super(TestTTSDataset, self).__init__(*args, **kwargs)
super().__init__(*args, **kwargs)
self.max_loader_iter = 4
self.ap = AudioProcessor(**c.audio)
+6 -6
View File
@@ -22,7 +22,7 @@ def test_torch_stft():
torch_stft = TorchSTFT(ap.fft_size, ap.hop_length, ap.win_length)
# librosa stft
wav = ap.load_wav(WAV_FILE)
M_librosa = abs(ap._stft(wav)) # pylint: disable=protected-access
M_librosa = abs(ap._stft(wav)) # pylint: disable=protected-access
# torch stft
wav = torch.from_numpy(wav[None, :]).float()
M_torch = torch_stft(wav)
@@ -42,9 +42,10 @@ def test_stft_loss():
def test_multiscale_stft_loss():
stft_loss = MultiScaleSTFTLoss([ap.fft_size//2, ap.fft_size, ap.fft_size*2],
[ap.hop_length // 2, ap.hop_length, ap.hop_length * 2],
[ap.win_length // 2, ap.win_length, ap.win_length * 2])
stft_loss = MultiScaleSTFTLoss(
[ap.fft_size // 2, ap.fft_size, ap.fft_size * 2],
[ap.hop_length // 2, ap.hop_length, ap.hop_length * 2],
[ap.win_length // 2, ap.win_length, ap.win_length * 2])
wav = ap.load_wav(WAV_FILE)
wav = torch.from_numpy(wav[None, :]).float()
loss_m, loss_sc = stft_loss(wav, wav)
@@ -53,6 +54,7 @@ def test_multiscale_stft_loss():
assert loss_sc < 1.0
assert loss_m + loss_sc > 0
def test_melgan_feature_loss():
feats_real = []
feats_fake = []
@@ -71,7 +73,6 @@ def test_melgan_feature_loss():
loss = loss_func(feats_fake, feats_real)
assert loss.item() <= 1.0
feats_real = []
feats_fake = []
@@ -89,4 +90,3 @@ def test_melgan_feature_loss():
loss_func = MelganFeatureLoss()
loss = loss_func(feats_fake, feats_real)
assert loss.item() == 0