sagemaker-studio-image-build-cli icon indicating copy to clipboard operation
sagemaker-studio-image-build-cli copied to clipboard

Add SKLearn and SparkML registries

Open athewsey opened this issue 3 years ago • 0 comments

Issue #, if available:

Stop-gap for missing containers mentioned in #13, but doesn't provide the requested CLI input feature

Description of changes:

Conditionally log in to the appropriate ECR accounts for public SKLearn and SparkML images in the current AWS_DEFAULT_REGION, as listed in the SageMaker docs here

I figured it would be better to add the extra registries conditionally by region, since there are so many?

Testing done:

I patched the updated buildspec template into my pip installed version of the library and it enabled me to successfully build the container I was stuck on - but of course haven't been able to fully go and validate the registries for the many different regions!

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

athewsey avatar Apr 20 '21 08:04 athewsey