Signed-off-by: Esteban Papp <81431996+amznestebanpapp@users.noreply.github.com>
The drain function was used only before the API gained the ability to wait on the completion of a graph. This is the correct way to "drain" the task executor of work. Signed-off-by: Jeremy Ong <jcong@amazon.com>
This commit also addresses all PR feedback Signed-off-by: Jeremy Ong <jcong@amazon.com>