Add Hugging Face DLC images to sagemaker pre-built ecr image

[philschmid]

What does this PR do?

This PR adds the Hugging Face DLCs to the aws_sagemaker_prebuilt_ecr_image data source. The Hugging Face DLCs include flavors for TensorFlow and PyTorch with Training and Inference Containers. I checked the region account mapping.

example

data "aws_sagemaker_prebuilt_ecr_image" "test" {
  repository_name = "huggingface-pytorch-inference"
  image_tag       = "1.7.1-transformers4.6.1-gpu-py36-cu110-ubuntu18.04"
}

I hope the .changelog is correct.

Community Note

• Please vote on this pull request by adding a 👍 reaction to the original pull request comment to help the community and maintainers prioritize this request
• Please do not leave "+1" or other comments that do not add relevant new information or questions, they generate extra noise for pull request followers and do not help prioritize the request

Output from acceptance testing:

$ AWS_PROFILE=hf-sm make testacc TESTS=TestAccSageMakerPrebuiltECRImageDataSource_basic PKG=sagemaker
==> Checking that code complies with gofmt requirements...
TF_ACC=1 go test ./internal/service/sagemaker/... -v -count 1 -parallel 20 -run='TestAccSageMakerPrebuiltECRImageDataSource_basic' -timeout 180m
=== RUN   TestAccSageMakerPrebuiltECRImageDataSource_basic
=== PAUSE TestAccSageMakerPrebuiltECRImageDataSource_basic
=== CONT  TestAccSageMakerPrebuiltECRImageDataSource_basic
--- PASS: TestAccSageMakerPrebuiltECRImageDataSource_basic (15.46s)
PASS
ok      github.com/hashicorp/terraform-provider-aws/internal/service/sagemaker  17.434s```

```bash
$ AWS_PROFILE=hf-sm make testacc TESTS=TestAccSageMakerPrebuiltECRImageDataSource_region PKG=sagemaker
==> Checking that code complies with gofmt requirements...
TF_ACC=1 go test ./internal/service/sagemaker/... -v -count 1 -parallel 20 -run='TestAccSageMakerPrebuiltECRImageDataSource_region' -timeout 180m
=== RUN   TestAccSageMakerPrebuiltECRImageDataSource_region
=== PAUSE TestAccSageMakerPrebuiltECRImageDataSource_region
=== CONT  TestAccSageMakerPrebuiltECRImageDataSource_region
--- PASS: TestAccSageMakerPrebuiltECRImageDataSource_region (15.36s)
PASS
ok      github.com/hashicorp/terraform-provider-aws/internal/service/sagemaker  17.370s

[github-actions]

Welcome @philschmid 👋

It looks like this is your first Pull Request submission to the Terraform AWS Provider! If you haven’t already done so please make sure you have checked out our CONTRIBUTING guide and FAQ to make sure your contribution is adhering to best practice and has all the necessary elements in place for a successful approval.

Also take a look at our FAQ which details how we prioritize Pull Requests for inclusion.

Thanks again, and welcome to the community! 😃

[ewbankkit]

LGTM 🚀.

% make testacc TESTS=TestAccSageMakerPrebuiltECRImageDataSource_ PKG=sagemaker
==> Checking that code complies with gofmt requirements...
TF_ACC=1 go test ./internal/service/sagemaker/... -v -count 1 -parallel 20 -run='TestAccSageMakerPrebuiltECRImageDataSource_' -timeout 180m
=== RUN   TestAccSageMakerPrebuiltECRImageDataSource_basic
=== PAUSE TestAccSageMakerPrebuiltECRImageDataSource_basic
=== RUN   TestAccSageMakerPrebuiltECRImageDataSource_region
=== PAUSE TestAccSageMakerPrebuiltECRImageDataSource_region
=== CONT  TestAccSageMakerPrebuiltECRImageDataSource_basic
=== CONT  TestAccSageMakerPrebuiltECRImageDataSource_region
--- PASS: TestAccSageMakerPrebuiltECRImageDataSource_basic (8.17s)
--- PASS: TestAccSageMakerPrebuiltECRImageDataSource_region (8.22s)
PASS
ok  	github.com/hashicorp/terraform-provider-aws/internal/service/sagemaker	11.618s

[ewbankkit]

@philschmid Thanks for the contribution 🎉 👏.

[github-actions]

This functionality has been released in v3.70.0 of the Terraform AWS Provider. Please see the Terraform documentation on provider versioning or reach out if you need any assistance upgrading.

For further feature requests or bug reports with this functionality, please create a new GitHub issue following the template. Thank you!

[philschmid]

Hey @ewbankkit,
I just noticed i forgot to add them to the documentation. I opened a new PR for it.

• Add Documentation for Hugging Face DLC images in sagemaker pre-built ecr image  #22263

[github-actions]

This functionality has been released in v3.71.0 of the Terraform AWS Provider. Please see the Terraform documentation on provider versioning or reach out if you need any assistance upgrading.

For further feature requests or bug reports with this functionality, please create a new GitHub issue following the template. Thank you!

[github-actions]

I'm going to lock this pull request because it has been closed for 30 days ⏳. This helps our maintainers find and focus on the active issues.
If you have found a problem that seems related to this change, please open a new issue and complete the issue template so we can capture all the details necessary to investigate further.