Problem with CheckpointConfig s3uri in Tensorflow estimator

Values from estimator not passing to the step function definition (JSON). Had to manually set it in json by editing the step function from console to make the training job work.

` "CheckpointConfig": {
"S3Uri": "s3://bucket_name/location/for/checkpoint/"
},`
S3Uri is taken from `checkpoint_s3_uri` right? but in my case this is not working. `CheckpointConfig `dictionary is not generated.

Thank you for taking a look at it.

Start by tracing the TensorFlow estimator's checkpoint_s3_uri into the generated Step Functions JSON, focusing on CheckpointConfig and its S3Uri field. Done means the estimator-generated definition includes the expected CheckpointConfig and S3Uri so the training job works without manual console edits.

Written by the indexing model from the issue text.