Does the Step Function Data Science SDK support SageMaker Processing Jobs similar to Batch Transform Jobs?

[aws]

Does the Step Function Data Science SDK support SageMaker Processing Jobs similar to Batch Transform Jobs?

[cfregly]

looking forward to this being released! looks like the code was just merged. fingers-crossed.

[shunjd]

Hi Chris,

We are currently investigating the options to integrate with the new SageMaker Processing Jobs. The AWS Step Functions (the service) needs to implement this new service integration first, then we can integrate it into the Python SDK.

Shunjia

[trevor-smith]

Please keep us updated on this!! I would love for this feature to exist. I use processing jobs in my ML workflow and would love to be able to use a step function to control my pipeline instead of using lambda functions that rely on s3 triggers based on a put operation. Thank you!

[ARDivekar]

@shunjd we have a use-case for this. Could you please let us know what would be a timeline for this integration?

[igoichuk]

@shunjd I am also interested in the processing step. Is there a way to submit SageMaker processing job using LambdaStep? I assume there should be created a lambda function that submits SageMaker processing jobs.

[dhawalkp]

Hi @cfregly Please refer to my new notebook demonstrating the new ProcessingStep here -
https://github.com/awslabs/amazon-sagemaker-examples/blob/master/step-functions-data-science-sdk/step_functions_mlworkflow_processing/step_functions_mlworkflow_scikit_learn_data_processing_and_model_evaluation.ipynb