typing on dsl.cof

pypyr · Oct 17, 2022 · a745d2b · a745d2b
1 parent 659f024
commit a745d2b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pypyr/steps/dsl/cof.py b/pypyr/steps/dsl/cof.py
@@ -19,7 +19,8 @@ def instruction_from_dict(config: Mapping,
                           name: str,
                           instruction_type: type[ControlOfFlowInstruction],
                           context_key: str,
-                          original_config: tuple[str, Mapping]):
+                          original_config: tuple[str, Mapping]
+                          ) -> ControlOfFlowInstruction:
     """Create a Call or Jump from input dict."""
     if isinstance(config, str):
         groups = [config]
@@ -69,7 +70,7 @@ def control_of_flow_instruction(
         name: str,
         instruction_type: type[ControlOfFlowInstruction],
         context: Context,
-        context_key: str):
+        context_key: str) -> None:
     """Run a control of flow instruction.
 
     The step config in the context dict looks like this: