From 3c6c8967f37d1757ed1fcac7b3350417c604164c Mon Sep 17 00:00:00 2001 From: JarbasAi Date: Sat, 7 Oct 2023 22:04:13 +0100 Subject: [PATCH] fix return type --- ovos_audio/transformers.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/ovos_audio/transformers.py b/ovos_audio/transformers.py index 8fc27c0..b6a1ff3 100644 --- a/ovos_audio/transformers.py +++ b/ovos_audio/transformers.py @@ -2,6 +2,7 @@ from ovos_config import Configuration from ovos_plugin_manager.dialog_transformers import find_dialog_transformer_plugins, find_tts_transformer_plugins from ovos_utils.log import LOG +from typing import Tuple class DialogTransformersService: @@ -53,7 +54,7 @@ def shutdown(self): except Exception as e: LOG.warning(e) - def transform(self, wav_file: str, context: dict = None, sess: Session = None) -> str: + def transform(self, dialog: str, context: dict = None, sess: Session = None) -> Tuple[str, dict]: """ Get transformed audio and context for the preceding audio @param dialog: str to be spoken @@ -75,7 +76,7 @@ def transform(self, wav_file: str, context: dict = None, sess: Session = None) - LOG.debug(f"{module.name}: {dialog}") except Exception as e: LOG.exception(e) - return dialog + return dialog, context class TTSTransformersService: @@ -133,7 +134,7 @@ def shutdown(self): except Exception as e: LOG.warning(e) - def transform(self, wav_file: str, context: dict = None, sess: Session = None) -> str: + def transform(self, wav_file: str, context: dict = None, sess: Session = None) -> Tuple[str, dict]: """ Get transformed audio and context for the preceding audio @param wav_file: str path for the TTS wav file @@ -155,4 +156,4 @@ def transform(self, wav_file: str, context: dict = None, sess: Session = None) - LOG.debug(f"{module.name}: {wav_file}") except Exception as e: LOG.exception(e) - return dialog + return wav_file, context