Hook to apply decorators

Description

Hook to apply decorators

Context

Decorators are useful to modify behaviors of multiple node functions. Kedro 0.18.0 will deprecate applying decorators by decorate method of kedro.pipeline.Pipeline and it is encouraged to use hooks to apply decorators, but there are no official hooks to apply decorators.

Possible Implementation

Provide hooks to apply decorators

Possible Alternatives

Continue to support decorators by kedro.pipeline.Pipeline even in Kedro 0.18.0 and later.

[AntonyMilneQB]

If you prefer chaining decorators together like in the existing Node.decorate notation, then you should be able to write a small helper function to do this. Something like:

def decorate(function: Callable, *decorators: Callable) -> Callable:
    for decorator in reversed(decorators):
        function = decorator(function)
    return function

class ProjectHooks:
    @hook_impl
    def before_node_run(self, node: Node):
        node.func = decorate(node.func, deco1, deco2, deco3)

This should achieve node.func = deco1(deco2(deco3(node.func))) without needing to explicitly do the nesting.

[Minyus]

Thanks @AntonyMilneQB . Yes, using decorators as functions like node.func = log_time(node.func) seems to work fine. To use multiple decorators, however, the decorators need to be nested like node.func = deco1(deco2(deco3(node.func))), and it doesn’t look like Kedro’s style though. Anyway, thanks for a workable suggestion!