Set the new_fields in copy_model_files as None by default

coqui-ai · Dec 20, 2021 · 2b2ceca · 2b2ceca
1 parent 34749f8
commit 2b2ceca
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/TTS/trainer.py b/TTS/trainer.py
@@ -202,7 +202,7 @@ def __init__(  # pylint: disable=dangerous-default-value
             os.makedirs(output_path, exist_ok=True)
 
         # copy training assets to the output folder
-        copy_model_files(config, output_path, new_fields=None)
+        copy_model_files(config, output_path)
 
         # init class members
         self.args = args
@@ -274,7 +274,7 @@ def __init__(  # pylint: disable=dangerous-default-value
                 self.config.num_languages = self.model.language_manager.num_languages
 
             # update config file
-            copy_model_files(self.config, self.output_path, None)
+            copy_model_files(self.config, self.output_path)
 
         # setup criterion
         self.criterion = self.get_criterion(self.model)

diff --git a/TTS/utils/io.py b/TTS/utils/io.py
@@ -26,7 +26,7 @@ def __init__(self, *args, **kwargs):
         self.__dict__ = self
 
 
-def copy_model_files(config: Coqpit, out_path, new_fields):
+def copy_model_files(config: Coqpit, out_path, new_fields=None):
     """Copy config.json and other model files to training folder and add
     new fields.