🔧 Fix pytest.

Author: dathudeptrai

tensorflow_tts/processor/base_processor.py

@@ -64,7 +64,9 @@ def __post_init__(self):
 
         # processor name. usefull to use it for AutoProcessor
         self._processor_name = type(self).__name__
-        self.eos_id = self.symbol_to_id[self.setup_eos_token()]
+
+        if self.setup_eos_token():
+            self.eos_id = self.symbol_to_id[self.setup_eos_token()]
 
     def __getattr__(self, name: str) -> Union[str, int]:
         if "_id" in name:  # map symbol to id

tensorflow_tts/processor/libritts.py

@@ -84,7 +84,7 @@ def get_one_sample(self, item):
         return sample
 
     def setup_eos_token(self):
-        return None
+        return None # because we do not use this 
 
     def text_to_sequence(self, text):
         if (

test/test_base_processor.py

@@ -21,6 +21,9 @@ def get_one_sample(self, item):
     def text_to_sequence(self, text):
         return ["0"]
 
+    def setup_eos_token(self):
+        return None
+
 
 @pytest.fixture
 def processor(tmpdir):