in lambda/build/lambda_pipeline_change.py [0:0]
def lambda_handler(event, context):
try:
logger.debug(json.dumps(event))
if (
event.get("source") == "aws.sagemaker"
and event.get("detail-type")
== "SageMaker Model Building Pipeline Execution Status Change"
):
status_code, result = update_pipeline_rules(event)
else:
raise Exception("Expect SageMaker Model Package State Change")
logger.debug(json.dumps(result))
return {"statusCode": status_code, "body": json.dumps(result)}
except ClientError as e:
error_message = e.response["Error"]["Message"]
logger.error(error_message)
raise Exception(error_message)
except Exception as e:
logger.error(e)
raise e