end_time

start_time

@task

from prefect import Flow
from prefect.tasks.core.function import FunctionTask
from time import sleep, time
from typing import Any, Callable

class timed_task(FunctionTask):
    def __init__(
        self,
        fn: Callable = None,
        name: str = None,
        **kwargs: Any
    ):
        def timed_fn(**kwargs) -> Callable:
            t1 = time()
            result = fn(**kwargs)
            t2 = time()
            print(f'Function {fn.__name__!r} executed in {(t2-t1):.4f}s')
            return result
        
        super().__init__(fn=timed_fn, name=name, **kwargs)

@timed_task
def my_func() -> None:
    sleep(10)
 
with Flow('My Timed Flow') as flow:
    my_func()

if __name__ == "__main__":
    flow.run()