Fix bos/eos test

rhasspy · Nov 12, 2021 · 3f30daa · 3f30daa
1 parent d9aa9ed
commit 3f30daa
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_phonemes2ids.py b/tests/test_phonemes2ids.py
@@ -114,7 +114,11 @@ def test_bos_eos(self):
         phoneme_to_id = {"a": 1, "b": 2, "c": 3, bos: 4, eos: 5}
 
         ids = phonemes2ids(
-            word_phonemes=word_phonemes, phoneme_to_id=phoneme_to_id, bos=bos, eos=eos
+            word_phonemes=word_phonemes,
+            phoneme_to_id=phoneme_to_id,
+            bos=bos,
+            eos=eos,
+            auto_bos_eos=True,
         )
 
         self.assertEqual(ids, [4, 1, 2, 3, 2, 3, 1, 5])