Skip to content

Feature Request: Support for Placeholders for Sagemaker Processing container arguments #94

Closed
@manshahe

Description

@manshahe

This is a feature request for the newly added stepfunctions.steps.sagemaker.ProcessingStep to accept placeholders for inputs to the container_argument parameter. An example case is depicted below:

execution_input = ExecutionInput(
    schema={
        "JobName": str,
        #This argument is for the Sagemaker Processing step
        "argument": str
    }
)

processing_step = ProcessingStep(
    job_name=execution_input["JobName"],
    processor={sagemaker processor object here},
    inputs=inputs,
    outputs = outputs,
    #Argument passed here to Sagemaker Processing
    container_arguments=['--argument', execution_input.get['argument'].to_jsonpath()],
    container_entrypoint=["python3", "/opt/ml/processing/input/code/scipt.py"]
)

Rationale: This will make the generated state machines more generic, we can generate 1 state machine and pass it arguments for different workloads. Without this functionality, I need to generate n state machines for a parameter which takes n possible values.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions