Apply isort and black reformatting

Signed-off-by: Kuray107 <Kuray107@users.noreply.github.com>
Kuray107 · Aug 15, 2024 · f7fc2f3 · f7fc2f3
1 parent 66a724e
commit f7fc2f3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/examples/audio/audio_to_audio_train.py b/examples/audio/audio_to_audio_train.py
@@ -51,7 +51,7 @@ class ModelType(str, Enum):
     Predictive = 'predictive'
     ScoreBased = 'score_based'
     SchroedingerBridge = 'schroedinger_bridge'
-    FlowMatching= 'flow_matching'
+    FlowMatching = 'flow_matching'
 
 
 def get_model_class(model_type: ModelType):

diff --git a/nemo/collections/audio/models/enhancement.py b/nemo/collections/audio/models/enhancement.py
@@ -631,6 +631,7 @@ class FlowMatchingAudioToAudioModel(AudioToAudioModel):
         - decoder: transforms sampler output into the time domain (synthesis transform)
         - ssl_pretrain_masking: if it is defined, perform the ssl pretrain masking for self reconstruction in the training process
     """
+
     def __init__(self, cfg: DictConfig, trainer: Trainer = None):
         super().__init__(cfg=cfg, trainer=trainer)
         self.sample_rate = self._cfg.sample_rate