diff --git a/layers/tacotron.py b/layers/tacotron.py index b5b6e132..329fdb11 100644 --- a/layers/tacotron.py +++ b/layers/tacotron.py @@ -399,8 +399,7 @@ class Decoder(nn.Module): self.memory_input = torch.cat([ new_memory, self.memory_input[:, :( self.memory_size - self.r) * self.memory_dim].clone() - ], - dim=-1) + ], dim=-1) else: # memory queue size smaller than number of frames per decoder iter self.memory_input = new_memory[:, :self.memory_size * self.memory_dim] diff --git a/tests/symbols_tests.py b/tests/symbols_tests.py index 9bec0f18..d35e887e 100644 --- a/tests/symbols_tests.py +++ b/tests/symbols_tests.py @@ -1,9 +1,8 @@ import unittest from utils.text import phonemes -from collections import Counter class SymbolsTest(unittest.TestCase): - def test_uniqueness(self): + def test_uniqueness(self): #pylint: disable=no-self-use assert sorted(phonemes) == sorted(list(set(phonemes))), " {} vs {} ".format(len(phonemes), len(set(phonemes))) \ No newline at end of file diff --git a/tests/test_loader.py b/tests/test_loader.py index dd23e530..9d151820 100644 --- a/tests/test_loader.py +++ b/tests/test_loader.py @@ -138,8 +138,8 @@ class TestTTSDataset(unittest.TestCase): # there is a slight difference between two matrices. # TODO: Check this assert cond more in detail. assert abs((abs(mel.T) - - abs(mel_dl[:-1]) - ).sum()) < 1e-5, (abs(mel.T)- abs(mel_dl[:-1])).sum() + - abs(mel_dl[:-1]) + ).sum()) < 1e-5, (abs(mel.T) - abs(mel_dl[:-1])).sum() # check mel-spec correctness mel_spec = mel_input[0].cpu().numpy()