From b7eefac47d1c701dc0540e729e9452cb2c00a4d4 Mon Sep 17 00:00:00 2001 From: Edresson Casanova Date: Mon, 14 Mar 2022 13:31:01 +0000 Subject: [PATCH] Rename set_ids_from_file to load_ids_from_file --- TTS/tts/utils/managers.py | 4 ++-- TTS/tts/utils/speakers.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/TTS/tts/utils/managers.py b/TTS/tts/utils/managers.py index 963226e0..03650d15 100644 --- a/TTS/tts/utils/managers.py +++ b/TTS/tts/utils/managers.py @@ -22,7 +22,7 @@ class BaseIDManager: self.ids = {} if id_file_path: - self.set_ids_from_file(id_file_path) + self.load_ids_from_file(id_file_path) @staticmethod def _load_json(json_file_path: str) -> Dict: @@ -43,7 +43,7 @@ class BaseIDManager: """ self.ids, _ = self.parse_ids_from_data(items) - def set_ids_from_file(self, file_path: str) -> None: + def load_ids_from_file(self, file_path: str) -> None: """Set IDs from a file. Args: diff --git a/TTS/tts/utils/speakers.py b/TTS/tts/utils/speakers.py index 79631c65..b02b7ccc 100644 --- a/TTS/tts/utils/speakers.py +++ b/TTS/tts/utils/speakers.py @@ -196,7 +196,7 @@ def get_speaker_manager(c: Coqpit, data: List = None, restore_path: str = None, speaker_manager.set_embeddings_from_file(speakers_file) elif not c.use_d_vector_file: # restor speaker manager with speaker ID file. speaker_ids_from_data = speaker_manager.ids - speaker_manager.set_ids_from_file(speakers_file) + speaker_manager.load_ids_from_file(speakers_file) assert all( speaker in speaker_manager.ids for speaker in speaker_ids_from_data ), " [!] You cannot introduce new speakers to a pre-trained model." @@ -207,7 +207,7 @@ def get_speaker_manager(c: Coqpit, data: List = None, restore_path: str = None, raise "use_d_vector_file is True, so you need pass a external speaker embedding file." elif c.use_speaker_embedding and "speakers_file" in c and c.speakers_file: # new speaker manager with speaker IDs file. - speaker_manager.set_ids_from_file(c.speakers_file) + speaker_manager.load_ids_from_file(c.speakers_file) if speaker_manager.num_speakers > 0: print(