Fix pylint issues

This commit is contained in:
Edresson 2021-08-15 13:44:58 -03:00 committed by Eren Gölge
parent ac9416fb86
commit 3fbbebd74d
5 changed files with 11 additions and 10 deletions

View File

@ -82,8 +82,8 @@ class VitsConfig(BaseTTSConfig):
add_blank (bool):
If true, a blank token is added in between every character. Defaults to `True`.
test_sentences (List[str]):
List of sentences to be used for testing.
test_sentences (List[List]):
List of sentences with speaker and language information to be used for testing.
Note:
Check :class:`TTS.tts.configs.shared_configs.BaseTTSConfig` for the inherited parameters.

View File

@ -837,5 +837,3 @@ class Vits(BaseTTS):
if eval:
self.eval()
assert not self.training

View File

@ -20,7 +20,6 @@ class LanguageManager:
>>> manager = LanguageManager(language_id_file_path=language_id_file_path)
>>> language_id_mapper = manager.language_ids
"""
num_languages: int = 0
language_id_mapping: Dict = {}
def __init__(
self,
@ -77,7 +76,6 @@ class LanguageManager:
file_path (str): Path to the target json file.
"""
self.language_id_mapping = self._load_json(file_path)
self.num_languages = len(self.language_id_mapping)
def save_language_ids_to_file(self, file_path: str) -> None:
"""Save language IDs to a json file.
@ -99,7 +97,7 @@ def _set_file_path(path):
return path_continue
return None
def get_language_manager(c: Coqpit, data: List = None, restore_path: str = None, out_path: str = None) -> LanguageManager:
def get_language_manager(c: Coqpit, data: List = None, restore_path: str = None) -> LanguageManager:
"""Initiate a `LanguageManager` instance by the provided config.
Args:

View File

@ -38,6 +38,11 @@ class TestTTSDataset(unittest.TestCase):
def _create_dataloader(self, batch_size, r, bgs):
items = ljspeech(c.data_path, "metadata.csv")
# add a default language because now the TTSDataset expect a language
language = ""
items = [[*item, language] for item in items]
dataset = TTSDataset(
r,
c.text_cleaner,