I have a state machine create with CDK (TypeScript). In this state machine I have a map state in which I run multiple ML model trainings with SageMaker, amount depends on the ou