Delayed setting of resource restrictions to tasks

[andrew-esteban-imc]

Greetings,

I have a workflow at the moment where there may be a race condition between Scheduler.update_graph and Scheduler.client_desires_keys. When update_graph occurs first, resource restrictions are correctly applied to the tasks through Scheduler._apply_annotations, however when client_desires_keys comes first, any tasks that were included in the call to client_desires_keys do not have annotations applied to them during update_graph, due to _apply_annotations only acting on tasks that did not already have a task state.

I can probably figure out a way to overcome this race condition on my end, but I was wondering if there was a way to have a delayed setting of annotations?