added return type on stack outputs

alan-turing-institute · Nov 18, 2024 · e3b663b · e3b663b
1 parent 2aef05b
commit e3b663b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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.