Skip to content

Commit f9f48cc

Browse files
author
Ji Chen
committed
add seed in get_whitenoise function
1 parent cf90566 commit f9f48cc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/test_datasets.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -127,13 +127,13 @@ class TestLibriTTS(TempDirMixin, TorchaudioTestCase):
127127
def setUpClass(cls):
128128
cls.root_dir = cls.get_base_temp_dir()
129129
base_dir = os.path.join(cls.root_dir, 'LibriTTS', 'train-clean-100')
130-
for utterance_id in cls.utterance_ids:
130+
for i, utterance_id in enumerate(cls.utterance_ids):
131131
filename = f'{"_".join(str(u) for u in utterance_id)}.wav'
132132
file_dir = os.path.join(base_dir, str(utterance_id[0]), str(utterance_id[1]))
133133
os.makedirs(file_dir, exist_ok=True)
134134
path = os.path.join(file_dir, filename)
135135

136-
data = get_whitenoise(sample_rate=8000, duration=6, n_channels=1, dtype='int16')
136+
data = get_whitenoise(sample_rate=8000, duration=6, n_channels=1, dtype='int16', seed=i)
137137
save_wav(path, data, 8000)
138138
cls.data.append(normalize_wav(data))
139139

0 commit comments

Comments
 (0)