diff --git a/src/arc_spice/variational_pipelines/RTC_variational_pipeline.py b/src/arc_spice/variational_pipelines/RTC_variational_pipeline.py index a45d51d..1e97b43 100644 --- a/src/arc_spice/variational_pipelines/RTC_variational_pipeline.py +++ b/src/arc_spice/variational_pipelines/RTC_variational_pipeline.py @@ -260,7 +260,7 @@ def stack_translator_sentence_metrics( return stacked def stack_variational_outputs( - self, var_output: dict[str, list[Any]] + self, var_output: dict[str, list[dict]] ) -> dict[str, list]: """ Similar to above but this stacks variational output dictinaries into lists