Wrappers for tasks

[kvnkho]

Hi Krishna,

I will show Prefect 1 code, but I think it should be very similar. You can check this:

Logging Full Traceback with Custom Task Decorator

This is an example code snippet that will log the full traceback of an error to Prefect Cloud. It is also a good guide how to make your own task decorator. import traceback from prefect import task, Flow from functools import partial, wraps from prefect.executors import LocalDaskExecutor def custom_task(func=None, **task_init_kwargs): if func is None: return partial(custom_task, **task_init_kwargs) @wraps(func) def safe_func(**kwargs): try: return func(…