Add type annotation to engine wrapper

3 years ago · 3df5b314f1
parent efe3af29d6
commit 3df5b314f1
1 changed files with 1 additions and 1 deletions
--- a/tts/engine_wrapper.py
+++ b/tts/engine_wrapper.py
@ -15,7 +15,7 @@ class TTSEngine:
        tts_function must take the arguments text and filepath.
    """

-    def __init__(self, tts_function: function, reddit_object, path: str = "assets/mp3", max_length: int = 50):
+    def __init__(self, tts_function: function, reddit_object: dict, path: str = "assets/mp3", max_length: int = 50):
        self.tts_function = tts_function
        self.reddit_object = reddit_object
        self.path = path