Update a model package stage and status example (boto3)
To update a model package stage and status, you will need to assume an
execution role with the relevant permissions. The following provides an example
on how you can update the stage status using the UpdateModelPackage API using AWS SDK for Python (Boto3).
In this example, the ModelLifeCycle stage
"Development" and stage status "Approved"
condition keys for the UpdateModelPackage API action has been granted to
the your execution role. You can include a description in
stage-description
- SageMaker Python SDK v3
from sagemaker.core.helper.session_helper import get_execution_role, session
import boto3
region = boto3.Session().region_name role = get_execution_role()
sm_client = boto3.client('sagemaker', region_name=region)
model_package_update_input_dict = {
"ModelLifeCycle" : {
"stage" : "Development",
"stageStatus" : "Approved",
"stageDescription" : "stage-description"
}
}
model_package_update_response = sm_client.update_model_package(**model_package_update_input_dict)
- SageMaker Python SDK v2 (Legacy)
from sagemaker import get_execution_role, session
import boto3
region = boto3.Session().region_name role = get_execution_role()
sm_client = boto3.client('sagemaker', region_name=region)
model_package_update_input_dict = {
"ModelLifeCycle" : {
"stage" : "Development",
"stageStatus" : "Approved",
"stageDescription" : "stage-description"
}
}
model_package_update_response = sm_client.update_model_package(**model_package_update_input_dict)
