3개 답변
- 최신
- 최다 투표
- 가장 많은 댓글
1
That is correct. Figured that. Looks like one to call that API is to have LambaStep
and chain it after RegisterModel
step . Here is lambda implementation. Feels a bit heavy workaround for this objective. So let me know if there is a better /lighter way to achieving the same. Thanks.
sagemaker_client = client = boto3.client("sagemaker")
def lambda_handler(event, context):
print(boto3.__version__)
logger.info(f"Received Event: {event}")
logger.info(f"Boto version: {boto3.__version__}")
model_arn = event['model_arn']
model_arn = 'arn:aws:sagemaker:us-east-1:ACCOUNT:model-package/model_package_name/version'
response = client.describe_model_package(
ModelPackageName=model_arn
)
help(client.update_model_package)
response = client.update_model_package(
ModelPackageArn=model_arn,
ModelApprovalStatus=response['ModelApprovalStatus'],
CustomerMetadataProperties={'string': 'string'},
)
return {"statusCode": 200, "body": json.dumps("Model metadata updated successfully")}
답변함 2년 전
The only problem with approach getting registered model ARN in SageMaker pipelines. It is buried somewhere in
RegisterModel
step, but I can't seem to figure out how to extract it. However, SM Studio does access it and shows in UI as step output.
0
Hi,
You can use the UpdateModelPackage API to update custom metadata properties on an already registered model package whether it's in a pipeline or not.
답변함 2년 전
관련 콘텐츠
- AWS 공식업데이트됨 2년 전
Perfect timing. This support got added 2 days ago! Thanks for pointing to it.