Bring legacy machine learning code into Amazon SageMaker using AWS Step Functions