style

Signed-off-by: Jason <[email protected]>

nemo/collections/tts/models/tacotron2.py

@@ -13,26 +13,26 @@
 # limitations under the License.
 
 from dataclasses import dataclass
-from typing import Dict, Optional, List
+from typing import Dict, List, Optional
 
 import torch
 from omegaconf import MISSING, DictConfig, OmegaConf, open_dict
 from torch import nn
 from torch.nn.functional import pad
 
 from nemo.collections.tts.helpers.helpers import get_mask_from_lengths, tacotron2_log_to_tb_func
-from nemo.core.classes import ModelPT, Loss, typecheck
-from nemo.utils import logging
+from nemo.core.classes import Loss, ModelPT, typecheck
 from nemo.core.neural_types.elements import (
-    LengthsType,
-    MelSpectrogramType,
-    LossType,
-    LogitsType,
     AudioSignal,
     EmbeddedTextType,
+    LengthsType,
+    LogitsType,
+    LossType,
+    MelSpectrogramType,
     SequenceToSequenceAlignmentType,
 )
 from nemo.core.neural_types.neural_type import NeuralType
+from nemo.utils import logging
 from nemo.utils.decorators import experimental
 
 

nemo/collections/tts/models/waveglow.py

@@ -20,18 +20,18 @@
 
 from nemo.collections.tts.helpers.helpers import waveglow_log_to_tb_func
 from nemo.collections.tts.modules.waveglow import OperationMode
-from nemo.core.classes import ModelPT, Loss, typecheck
-from nemo.utils import logging
-from nemo.utils.decorators import experimental
+from nemo.core.classes import Loss, ModelPT, typecheck
 from nemo.core.neural_types.elements import (
-    NormalDistributionSamplesType,
-    VoidType,
-    LossType,
-    LengthsType,
     AudioSignal,
+    LengthsType,
+    LossType,
     MelSpectrogramType,
+    NormalDistributionSamplesType,
+    VoidType,
 )
 from nemo.core.neural_types.neural_type import NeuralType
+from nemo.utils import logging
+from nemo.utils.decorators import experimental
 
 
 class WaveGlowLoss(Loss):

nemo/collections/tts/modules/tacotron2.py

@@ -16,14 +16,6 @@
 from torch.autograd import Variable
 from torch.nn import functional as F
 
-_NATIVE_AMP = False
-try:
-    from torch.cuda.amp import autocast
-
-    _NATIVE_AMP = True
-except ImportError:
-    pass
-
 from nemo.collections.tts.helpers.helpers import get_mask_from_lengths
 from nemo.collections.tts.modules.submodules import Attention, ConvNorm, LinearNorm, Prenet
 from nemo.core.classes import NeuralModule, typecheck
@@ -38,6 +30,14 @@
 from nemo.utils import logging
 from nemo.utils.decorators import experimental
 
+_NATIVE_AMP = False
+try:
+    from torch.cuda.amp import autocast
+
+    _NATIVE_AMP = True
+except ImportError:
+    pass
+
 
 @experimental  # TODO: Need to implement abstratct methods: save_to, restore_from, but how?
 class Encoder(NeuralModule):