This is the second state for the AWS Step Function to automate training, build and deployment of an Amazon SageMaker model

stepfunctions-state-02.json

"Create Model": {
      "Parameters": {
        "PrimaryContainer": {
          "Image": "544295431143.dkr.ecr.ap-southeast-2.amazonaws.com/xgboost:1",
          "Environment": {},
          "ModelDataUrl.$": "$.ModelArtifacts.S3ModelArtifacts"
        },
        "ExecutionRoleArn": "iam-role-arn",
        "ModelName.$": "$.TrainingJobName"
      },
      "Resource": "arn:aws:states:::sagemaker:createModel",
      "Type": "Task",
      "ResultPath":"$.taskresult",
      "Next": "Create Endpoint Config"
    },