From 8cdd423234317d1d3698f215da47cb1743e8f99e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Tue, 25 May 2021 14:41:13 +0200
Subject: [PATCH] styling formatting.py

---
 TTS/utils/arguments.py | 39 ++++++++++++++++++++++++++-------------
 1 file changed, 26 insertions(+), 13 deletions(-)

diff --git a/TTS/utils/arguments.py b/TTS/utils/arguments.py
index 9d92ae82..55bad4f2 100644
--- a/TTS/utils/arguments.py
+++ b/TTS/utils/arguments.py
@@ -29,16 +29,16 @@ def init_arguments(argv):
     parser.add_argument(
         "--continue_path",
         type=str,
-        help=(
-            "Training output folder to continue training. Used to continue "
-            "a training. If it is used, 'config_path' is ignored."
-        ),
+        help=("Training output folder to continue training. Used to continue "
+              "a training. If it is used, 'config_path' is ignored."),
         default="",
         required="--config_path" not in argv,
     )
     parser.add_argument(
-        "--restore_path", type=str, help="Model file to be restored. Use to finetune a model.", default=""
-    )
+        "--restore_path",
+        type=str,
+        help="Model file to be restored. Use to finetune a model.",
+        default="")
     parser.add_argument(
         "--best_path",
         type=str,
@@ -48,12 +48,23 @@ def init_arguments(argv):
         ),
         default="",
     )
+    parser.add_argument("--config_path",
+                        type=str,
+                        help="Path to config file for training.",
+                        required="--continue_path" not in argv)
+    parser.add_argument("--debug",
+                        type=bool,
+                        default=False,
+                        help="Do not verify commit integrity to run training.")
     parser.add_argument(
-        "--config_path", type=str, help="Path to config file for training.", required="--continue_path" not in argv
-    )
-    parser.add_argument("--debug", type=bool, default=False, help="Do not verify commit integrity to run training.")
-    parser.add_argument("--rank", type=int, default=0, help="DISTRIBUTED: process rank for distributed training.")
-    parser.add_argument("--group_id", type=str, default="", help="DISTRIBUTED: process group id.")
+        "--rank",
+        type=int,
+        default=0,
+        help="DISTRIBUTED: process rank for distributed training.")
+    parser.add_argument("--group_id",
+                        type=str,
+                        default="",
+                        help="DISTRIBUTED: process group id.")
 
     return parser
 
@@ -148,7 +159,8 @@ def process_args(args):
         print("   >  Mixed precision mode is ON")
     experiment_path = args.continue_path
     if not experiment_path:
-        experiment_path = create_experiment_folder(config.output_path, config.run_name, args.debug)
+        experiment_path = create_experiment_folder(config.output_path,
+                                                   config.run_name, args.debug)
     audio_path = os.path.join(experiment_path, "test_audios")
     # setup rank 0 process in distributed training
     tb_logger = None
@@ -169,7 +181,8 @@ def process_args(args):
         os.chmod(experiment_path, 0o775)
         tb_logger = TensorboardLogger(experiment_path, model_name=config.model)
         # write model desc to tensorboard
-        tb_logger.tb_add_text("model-config", f"<pre>{config.to_json()}</pre>", 0)
+        tb_logger.tb_add_text("model-config", f"<pre>{config.to_json()}</pre>",
+                              0)
     c_logger = ConsoleLogger()
     return config, experiment_path, audio_path, c_logger, tb_logger