Fix typing for callables

bit-bots · Apr 5, 2024 · 43e637b · 43e637b
1 parent c008a76
commit 43e637b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dynamic_stack_decider/dynamic_stack_decider/sequence_element.py b/dynamic_stack_decider/dynamic_stack_decider/sequence_element.py
@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Callable
 
 from dynamic_stack_decider.abstract_action_element import AbstractActionElement
 from dynamic_stack_decider.abstract_stack_element import AbstractStackElement
@@ -22,7 +22,7 @@ def __init__(
         blackboard,
         dsd: "DSD",
         actions: list[ActionTreeElement],
-        init_function: callable[[AbstractTreeElement], AbstractStackElement],
+        init_function: Callable[[AbstractTreeElement], AbstractStackElement],
     ):
         """
         :param blackboard: Shared blackboard for data exchange and code reuse between elements