Step functions, EndpointStep, wait_for_completion?

Hi,

I was wondering why EndpointStep does not have wait_for_completion parameter like TrainingStep?
I am using Lambda function after EndpointStep to test the endoint and now I have to use WaitStep and Retry in Lambda testfunction to make sure the endpoint is already there once testing the endpoint with Lambda starts.

This feels dumm and I wish there was some way to have wait_for_completion-like functionality in the EndpointStep

Compare the EndpointStep behavior described here with the existing TrainingStep wait_for_completion functionality, then trace how the Lambda test function is invoked after EndpointStep. Done should let a downstream Lambda test start only after the endpoint is ready, without requiring a separate WaitStep and Retry sequence.

Written by the indexing model from the issue text.