Something like `NodeType: Union[BaseNode, Callable[[Any, Any], State]]`
Something like
NodeType: Union[BaseNode, Callable[[Any, Any], State]]