enforce list append semantic; prevents numpy add

pull/10/head
Thomas Werkmeister 2019-07-26 13:40:58 +02:00
parent 1edd9e1f3d
commit 215eb014ca
2 changed files with 2 additions and 1 deletions

View File

@ -102,6 +102,7 @@ class MyDataset(Dataset):
cache_path)
if self.enable_eos_bos:
phonemes = pad_with_eos_bos(phonemes)
phonemes = np.asarray(phonemes, dtype=np.int32)
return phonemes

View File

@ -47,7 +47,7 @@ def text2phone(text, language):
def pad_with_eos_bos(phoneme_sequence):
return [_PHONEMES_TO_ID[_bos]] + phoneme_sequence + [_PHONEMES_TO_ID[_eos]]
return [_PHONEMES_TO_ID[_bos]] + list(phoneme_sequence) + [_PHONEMES_TO_ID[_eos]]
def phoneme_to_sequence(text, cleaner_names, language, enable_eos_bos=False):