Fix format problems

chintu619 · Apr 24, 2022 · be0112b · be0112b
1 parent 16acdad
commit be0112b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/espnet2/enh/separator/svoice_separator.py b/espnet2/enh/separator/svoice_separator.py
@@ -147,8 +147,8 @@ def __init__(
         )
 
     def forward(
-        self, 
-        input: torch.Tensor, 
+        self,
+        input: torch.Tensor,
         ilens: torch.Tensor,
         additional: Optional[Dict] = None,
     ) -> Tuple[List[torch.Tensor], torch.Tensor, OrderedDict]:

diff --git a/espnet2/tasks/enh.py b/espnet2/tasks/enh.py
@@ -99,7 +99,10 @@
 loss_wrapper_choices = ClassChoices(
     name="loss_wrappers",
     classes=dict(
-        pit=PITSolver, fixed_order=FixedOrderSolver, multilayer_pit=MultiLayerPITSolver, dpcl=DPCLSolver
+        pit=PITSolver,
+        fixed_order=FixedOrderSolver,
+        multilayer_pit=MultiLayerPITSolver,
+        dpcl=DPCLSolver,
     ),
     type_check=AbsLossWrapper,
     default=None,