Open
Description
What would you like to be added:
As we are introducing wait flag for Slurm mode, this feature can be beneficial for other modes as well.
Why is this needed:
This is a follow-up of #40. See #40 (comment).
Completion requirements:
This enhancement requires the following artifacts:
- Design doc
- API change
- Docs update
The artifacts should be linked in subsequent comments.