When using step decorator, the step type is fixed to StepTypeEnum.TRAINING

[RaydelMiranda]

I am creating a pipeline using the step decorator, as already suggested in the title of this discussion. However, when visualizing the pipeline in SageMaker studio, all my steps fall in the Training Category, when I am doing also preprocessing and deployment.

After a quick look at the code, I found out there is no way to set the desired step type while using the step decorator:

I would like to know the reason behind this and also if there is a workaround that would allow me to create, for instance, a processing step (using the decorator) that would actually be shown as a processing step in the Sage maker console.

Thanks in advance!