Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/biomapas/b.cfnsagemakerendpoint
AWS CDK resource that handles the deployment and update of SageMaker models' endpoint any time the code changes in your S3 bucket.
https://github.com/biomapas/b.cfnsagemakerendpoint
aws aws-cdk aws-cdk-python aws-cloudformation sagemaker sagemaker-deployment sagemaker-endpoint
Last synced: 9 days ago
JSON representation
AWS CDK resource that handles the deployment and update of SageMaker models' endpoint any time the code changes in your S3 bucket.
- Host: GitHub
- URL: https://github.com/biomapas/b.cfnsagemakerendpoint
- Owner: Biomapas
- License: apache-2.0
- Created: 2021-12-14T11:17:49.000Z (about 3 years ago)
- Default Branch: master
- Last Pushed: 2021-12-21T14:33:09.000Z (about 3 years ago)
- Last Synced: 2024-11-06T12:55:26.380Z (about 2 months ago)
- Topics: aws, aws-cdk, aws-cdk-python, aws-cloudformation, sagemaker, sagemaker-deployment, sagemaker-endpoint
- Language: Python
- Homepage:
- Size: 39.1 KB
- Stars: 0
- Watchers: 2
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- Changelog: HISTORY.md
- License: LICENSE
Awesome Lists containing this project
README
# 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](https://biomapas.com) 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](https://docs.aws.amazon.com/sagemaker/latest/dg/whatis.html)
- [AWS SDK for Python (Boto 3)](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html?icmpid=docs_sagemaker_lp)
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.
```bash
pip install .
```
Or
```bash
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:
```python
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:
```json
{
"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``:
```python
example_endpoint_config_props = CfnEndpointConfigProps(...)
```
3. Configure SageMaker endpoint properties as ``CfnEnpointProps``:
```python
example_endpoint_props = CfnEndpointProps(...)
```
4. Finally, setup ``B.CfnSagemakerEndpoint``'s ``SagemakerEndpoint`` resource using previously defined
properties:
```python
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.