b-cfn-sagemaker-endpoint 0.0.4

AWS CloudFormation resource that handles the deployment and updates of SageMaker models endpoint.

B.CfnSagemakerEndpoint

b-cfn-sagemaker-endpoint - AWS CloudFormation resource that handles the deployment and update of SageMaker models endpoint.

Description

This resource handles the deployment and update of SageMaker models endpoint. It is designed to enable automatic update of SageMaker's models endpoint in the event of modifying the source model data. This is achieved by utilizing S3's' event notifications. On updating the target S3 bucket objects, an event is emitted that is handled by a lambda function which updates the deployed SageMaker endpoint.

Design

This resource implements architecture showed in the following UML diagram:

Endpoint update is done via boto3's function update_endpoint() that requires a new configuration to be provided. Instead of creating a new instance of it each time, two identical configurations (A & B) with different names are created only once, during the deployment. Each update_endpoint() call effectively swaps them together, allowing the endpoint to be refreshed with new up-to-date source model(-s) data. This can be seen in the UML activity diagram below:

Remarks

Biomapas aims to modernise life-science industry by sharing its IT knowledge with other companies and the community.

Related technology

• Python >= 3.8
• Amazon Web Services (AWS)
• Amazon SageMaker

Assumptions

The project assumes that the person working with it have basic knowledge in python programming.

Useful sources

SageMaker documentation:

• Developer guide
• AWS SDK for Python (Boto 3)

See code documentation for any additional sources and references.

Install

Use the package manager pip to install this package. This project is not in the PyPi repository yet. Install directly from source or PyPI.

pip install .

Or

pip install b-cfn-sagemaker-endpoint

Usage & Examples

This resource handles aws-cdk.aws-sagemaker library's resources: CnfModel, CnfEndpointConfig & CnfEndpoint deployment. The user is required only to provide configurations for each of these items via their properties, i.e.: CnfModelProps -> CnfModel, that on themselves do not create any resources.

To deploy a SageMaker model(-s) endpoint the following steps have to be taken:

1. Configure SageMaker model properties as a ModelProps class object:

   example_model_props = ModelProps(
       props=CfnModelProps(...),
       custom_id=..., # (optional) If not provided, it is generated automatically.
   )
   

   Note: CfnModelProps require an execution role ARN. This role must have sagemaker.amazonaws.com set as the trusted entity that can assume it. In usual cases the following permissions are required:

   {
       "Version": "2012-10-17",
       "Statement": [
           {
               "Action": [
                   "cloudwatch:PutMetricData",
                   "ecr:GetAuthorizationToken",
                   "ecr:BatchCheckLayerAvailability",
                   "ecr:GetDownloadUrlForLayer",
                   "ecr:BatchGetImage",
                   "logs:CreateLogStream",
                   "logs:PutLogEvents",
                   "logs:CreateLogGroup",
                   "logs:DescribeLogStreams"
               ],
               "Resource": "*",
               "Effect": "Allow"
           },
           {
               "Action": [
                   "s3:PutObject",
                   "s3:ListBucket",
                   "s3:GetObject",
                   "s3:DeleteObject"
               ],
               "Resource": "*",
               "Effect": "Allow"
           }
       ]
   }
   

2. Setup SageMaker endpoint configuration properties as CfnEnpointConfigProps:

   example_endpoint_config_props = CfnEndpointConfigProps(...)
   

3. Configure SageMaker endpoint properties as CfnEnpointProps:

   example_endpoint_props = CfnEndpointProps(...)
   

4. Finally, setup B.CfnSagemakerEndpoint's SagemakerEndpoint resource using previously defined properties:

   SagemakerEndpoint(
       scope=...,
       id='your-scoped-cdk-resource-id',
       endpoint_props=example_endpoint_props,
       endpoint_config_props=example_endpoint_config_props,
       models_props=[
           example_model_props
       ],
       models_bucket=Bucket(...),
       # (Optional) If no explicit bucket events are provided, `EventType.OBJECT_CREATED` 
       # events are emitted for all bucket objects.
       bucket_events=[
           BucketEvent(EventType.OBJECT_CREATED, [
               NotificationKeyFilter(
                   prefix=...,
                   # (Optional) Setting suffix, specifies on which files changes, based on 
                   # file type, should endpoint be updated. This is useful if updates are 
                   # required only for SageMaker model objects, i.e.: "model.tar.gz". 
                   # By default, if no bucket events are explicitly provided suffix is set 
                   # to ".tar.gz".
                   suffix='.tar.gz'
               )
           ])
       ],
       # (Optional) Set this value to the max time (in seconds) for how long it will take 
       # to upload updated contents to the S3 bucket. By default it is set to 60 seconds.
       wait_time=...
   )
   

Once, SagemakerEndpoint is deployed, any changes in the source models bucket specified by bucket_events, starts endpoint update/refresh. During this time, it's status becomes "Updating" & no further update calls are handled.

Known limits

Changing settings of CfnModel or CfnEndpointConfig resources, does not update the CfnEndpoint resource itself. To achieve this functionality, likely, a low-level CustomResource implementation would be required.

Testing

For now, a single test is implemented to make sure that the SageMaker endpoint is automatically updated with the latest model data found in the source S3 bucket.

Contribution

Found a bug? Want to add or suggest a new feature? Contributions of any kind are gladly welcome. Contact your direct supervisor, create a pull-request or an issue in Jira platform.

Release history

0.0.4

• Updated README with architecture & activity diagrams.

0.0.3

• Updated README.

0.0.2

• Added testing of deployed SageMaker endpoint update/refresh feature.
• Improved project documentation.
• Improved code quality.
• Fixed GitHub pipeline.

0.0.1

• Initial implementation of B.CfnSagemakerEndpoint resource.