How do I handle job time out errors within the task?

[epicwhale]

I have a task function which has a timeout explicitly set. When it times out I notice it throws the below exception(s) trace... and I am struggling to figure out where/how to handle it gracefully?

3.00s ! d786aa9ab9484760ada98578227f0bd3:example.task failed, TimeoutError: 
...

asyncio.exceptions.CancelledError
...

    raise TimeoutError from exc_val
                            └ CancelledError()

TimeoutError

I need to do some clean up of resources when such a timeout happens, but I need access to the initial parameters/args with which the task was called to do so. How do I handle this scenario?

If I register an on_job_end handle for my worker, the ctx in it does not receive the job information.

Any pointers would be appreciated!