From 4df31f7fbd65ff186d2700cd1b343f88f32110c0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Tue, 18 May 2021 14:50:03 +0200
Subject: [PATCH] unused_speakers argument for ignoring speaker ids in
 multi-speaker training

---
 TTS/config/shared_configs.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/TTS/config/shared_configs.py b/TTS/config/shared_configs.py
index 94e1c6f3..3aa80c98 100644
--- a/TTS/config/shared_configs.py
+++ b/TTS/config/shared_configs.py
@@ -143,9 +143,11 @@ class BaseDatasetConfig(Coqpit):
             Dataset name that defines the preprocessor in use. Defaults to None.
         path (str):
             Root path to the dataset files. Defaults to None.
-        meta_file_train (Union[str, List]):
+        meta_file_train (str):
             Name of the dataset meta file. Or a list of speakers to be ignored at training for multi-speaker datasets.
             Defaults to None.
+        unused_speakers (List):
+            List of speakers IDs that are not used at the training. Default None.
         meta_file_val (str):
             Name of the dataset meta file that defines the instances used at validation.
         meta_file_attn_mask (str):
@@ -155,9 +157,8 @@ class BaseDatasetConfig(Coqpit):
 
     name: str = ""
     path: str = ""
-    meta_file_train: Union[
-        str, List
-    ] = ""  # TODO: don't take ignored speakers for multi-speaker datasets over this. This is Union for SC-Glow compat.
+    meta_file_train: str = ""
+    ununsed_speakers: List[str] = None
     meta_file_val: str = ""
     meta_file_attn_mask: str = ""