Saad Bin Akhlaq
04/22/2024, 2:34 PMSaad Bin Akhlaq
04/22/2024, 2:35 PMNate
04/22/2024, 3:09 PMreturn_state
is useful for this
In [1]: from prefect import task, flow
In [2]: @task
...: def terrible():
...: raise ValueError('oops')
...:
In [3]: @flow
...: def f():
...: state = terrible(return_state=True)
...: assert state.is_failed()
...: return "and its ok"
10:09:26.182 | ERROR | Task run 'terrible-0' - Finished in state Failed('Task run encountered an exception ValueError: oops')
10:09:26.300 | INFO | Flow run 'masterful-wrasse' - Finished in state Completed()
Out[4]: 'and its ok'
Saad Bin Akhlaq
04/23/2024, 6:29 AM