aws / amazon-sagemaker-examples

Example 📓 Jupyter notebooks that demonstrate how to build, train, and deploy machine learning models using 🧠 Amazon SageMaker.
https://sagemaker-examples.readthedocs.io
Apache License 2.0
10.16k stars 6.78k forks source link

fix: use correct client for create env client, fix environment creation #4787

Closed danabens closed 6 days ago

danabens commented 6 days ago

Issue #, if available: N/A

Description of changes: The CreateEnvironmentAction API call uses a parameter that is only available in the byod DataZone model. Specifically parameters: "sageMaker": {}. Use the byod model to call CreateEnvironmentAction.

Testing done: Verified using latest public DataZone API model and then ran through script e2e in CMH prod.

Merge Checklist

Put an x in the boxes that apply. You can also fill these out after creating the PR. If you're unsure about any of them, don't hesitate to ask. We're here to help! This is simply a reminder of what we are going to look for before merging your pull request.

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.