aws-samples / sagemaker-studio-image-build-cli

CLI for building Docker images in SageMaker Studio using AWS CodeBuild.
https://pypi.org/project/sagemaker-studio-image-build/
MIT No Attribution
57 stars 25 forks source link

[Feature Request] Add support for cache usage #29

Open leitouran opened 10 months ago

leitouran commented 10 months ago

It would be great if we could manipulate CodeBuild create_project's cache argument so that builds take significantly less time. This is specially painful when developing container, where minor corrections/iterations can take 5-10 mins each.

CodeBuild supports different options for cache usage that can be indicated as an argument (cache) in the create_project method of the boto3 codebuild client (https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/codebuild/client/create_project.html)

...
    cache={
        'type': 'NO_CACHE'|'S3'|'LOCAL',
        'location': 'string',
        'modes': [
            'LOCAL_DOCKER_LAYER_CACHE'|'LOCAL_SOURCE_CACHE'|'LOCAL_CUSTOM_CACHE',
        ]
    },
...

We could leverage this argument to further optimize codebuild integration with sagemaker.