PrefectHQ / prefect

Prefect is a workflow orchestration tool empowering developers to build, observe, and react to data pipelines
https://prefect.io
Apache License 2.0
15.29k stars 1.5k forks source link

acquire/release token on TaskWorker without errors #14084

Closed jakekaplan closed 1 week ago

jakekaplan commented 1 week ago

When acquiring or releasing a token we need to be sure to not propagate RuntimeErrors so that the TaskWorker keeps running.

jakekaplan commented 1 week ago

Can you add a test for this? Submitting the same task run twice via execute_task_run should give us the confidence we need.

Added! Let me know if its not what you had in mind