It doesn't look like it's possible to specify the output path for the model artifacts in the TrainingStep class. You can only use the default output directory or define it in the estimator object before you pass it to the TrainingStep. In either case it doesn't look like its possible to set the output path from the execution input or the output of a prior task.