Being able to set "Created By" for a Sagemaker pipeline exectuion

[hossein-jazayeri]

Describe the feature

Right now the only way to set Created By field for a Sagemaker pipeline execution is through the Sagemaker studio console.
It would be very useful to be to set it through boto3 client as well!

Use Case

When a pipeline is triggered through boto3, we can't know who triggered the pipeline!

Proposed Solution

I can think of 2 scenarios: first, when an execution is started:

import boto3

boto3.client('sagemaker').start_pipeline_execution(
    PipelineName=..., 
    PipelineParameters=...
    CreatedBy= {
        'UserProfileArn': ..., 
    }
)

Or updating the existing execution:

import boto3

boto3.client('sagemaker').update_pipeline_execution(
    PipelineExecutionArn=..., 
    CreatedBy= {
        'UserProfileArn': ..., 
    }
)

Other Information

No response

Acknowledgements

• I may be able to implement this feature request
• This feature might incur a breaking change

SDK version used

1.26.25

Environment details (OS name and version, etc.)

Python/3.8.8 Linux/5.10.102.1-microsoft-standard-WSL2 exe/x86_64.ubuntu.20 prompt/off

[tim-finnigan]

Hi @hossein-jazayeri thanks for reaching out. Can't you use the describe_pipeline_execution command to get this information? There is a CreatedBy value documented in the response syntax.

[hossein-jazayeri]

Hi @tim-finnigan
describe_pipeline_execution returns the information, however, I'd like to set the value of CreatedBy!

[tim-finnigan]

Hi @hossein-jazayeri I see thanks for clarifying that. This would be a feature request for the Sagemaker team in that case, as they own the underlying StartPipelineExecution API which corresponds to the boto3 command. I created an issue in our shared SDK repository for this request, since the APIs are used across SDKs: aws/aws-sdk#540. I reached out to the Sagemaker team for feedback and will update that issue when I have more information. Please refer to that issue for updates going forward.