Address even more lint problems

vizay08 · Jul 19, 2019 · e3f8417 · e3f8417
1 parent 057a50a
commit e3f8417
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 3 deletions.
diff --git a/models/tacotron2.py b/models/tacotron2.py
@@ -88,8 +88,8 @@ def inference_truncated(self, text, speaker_ids=None):
 
     def _add_speaker_embedding(self, encoder_outputs, speaker_ids):
         if hasattr(self, "speaker_embedding") and speaker_ids is None:
-            raise RuntimeError(" [!] Model has speaker embedding layer but speaker_id is not provided")            
-        elif hasattr(self, "speaker_embedding") and speaker_ids is not None:
+            raise RuntimeError(" [!] Model has speaker embedding layer but speaker_id is not provided")
+        if hasattr(self, "speaker_embedding") and speaker_ids is not None:
             speaker_embeddings = self.speaker_embedding(speaker_ids)
 
             speaker_embeddings.unsqueeze_(1)

diff --git a/tests/test_layers.py b/tests/test_layers.py
@@ -5,6 +5,7 @@
 from layers.losses import L1LossMasked
 from utils.generic_utils import sequence_mask
 
+#pylint: disable=unused-variable
 
 class PrenetTests(unittest.TestCase):
     def test_in_out(self):

diff --git a/tests/test_loader.py b/tests/test_loader.py
@@ -8,6 +8,8 @@
 from datasets import TTSDataset
 from datasets.preprocess import ljspeech
 
+#pylint: disable=unused-variable
+
 file_path = os.path.dirname(os.path.realpath(__file__))
 OUTPATH = os.path.join(file_path, "outputs/loader_tests/")
 os.makedirs(OUTPATH, exist_ok=True)

diff --git a/tests/test_tacotron2_model.py b/tests/test_tacotron2_model.py
@@ -10,6 +10,8 @@
 from layers.losses import MSELossMasked
 from models.tacotron2 import Tacotron2
 
+#pylint: disable=unused-variable
+
 torch.manual_seed(1)
 use_cuda = torch.cuda.is_available()
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

diff --git a/tests/test_tacotron_model.py b/tests/test_tacotron_model.py
@@ -9,6 +9,8 @@
 from layers.losses import L1LossMasked
 from models.tacotron import Tacotron
 
+#pylint: disable=unused-variable
+
 torch.manual_seed(1)
 use_cuda = torch.cuda.is_available()
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
@@ -45,7 +47,7 @@ def test_train_step(self):
         criterion_st = nn.BCEWithLogitsLoss().to(device)
         model = Tacotron(
             32,
-            5, 
+            5,
             linear_dim=c.audio['num_freq'],
             mel_dim=c.audio['num_mels'],
             r=c.r,