From e3b663bddc4229599f038e60726da27e1429b49c Mon Sep 17 00:00:00 2001 From: J-Dymond Date: Mon, 18 Nov 2024 11:29:26 +0000 Subject: [PATCH] added return type on stack outputs --- .../variational_pipelines/RTC_variational_pipeline.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/arc_spice/variational_pipelines/RTC_variational_pipeline.py b/src/arc_spice/variational_pipelines/RTC_variational_pipeline.py index a2d33c3..a45d51d 100644 --- a/src/arc_spice/variational_pipelines/RTC_variational_pipeline.py +++ b/src/arc_spice/variational_pipelines/RTC_variational_pipeline.py @@ -259,7 +259,9 @@ def stack_translator_sentence_metrics( ] return stacked - def stack_variational_outputs(self, var_output: dict[str, list[Any]]): + def stack_variational_outputs( + self, var_output: dict[str, list[Any]] + ) -> dict[str, list]: """ Similar to above but this stacks variational output dictinaries into lists under a single key.