From c637aa04a2602438d145aa5745ac8f3bf057399b Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Fri, 16 Aug 2019 15:49:12 +0200 Subject: [PATCH] pylint --- layers/tacotron.py | 3 +-- tests/symbols_tests.py | 3 +-- tests/test_loader.py | 4 ++-- 3 files changed, 4 insertions(+), 6 deletions(-) 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()