Skip to content

Add framework_version to all TensorFlowModel examples #5038

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
merged 2 commits into from
Feb 17, 2025
Merged

Add framework_version to all TensorFlowModel examples #5038

merged 2 commits into from
Feb 17, 2025

Conversation

pintaoz-aws
Copy link
Contributor

@pintaoz-aws pintaoz-aws commented Feb 14, 2025 •
edited
Loading

Issue #, if available:
#2630
Description of changes:
Add framework_version to all TensorFlowModel examples
Testing done:

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.

General

  • I have read the CONTRIBUTING doc
  • I certify that the changes I am introducing will be backward compatible, and I have discussed concerns about this, if any, with the Python SDK team
  • I used the commit message format described in CONTRIBUTING
  • I have passed the region in to all S3 and STS clients that I've initialized as part of this change.
  • I have updated any necessary documentation, including READMEs and API docs (if appropriate)

Tests

  • I have added tests that prove my fix is effective or that my feature works (if appropriate)
  • I have added unit and/or integration tests as appropriate to ensure backward compatibility of the changes
  • I have checked that my tests are not configured for a specific region or account (if appropriate)
  • I have used unique_name_from_base to create resource names in integ tests (if appropriate)
  • If adding any dependency in requirements.txt files, I have spell checked and ensured they exist in PyPi

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

@pintaoz-aws pintaoz-aws requested a review from a team as a code owner February 14, 2025 00:04
benieric
benieric reviewed Feb 14, 2025
View reviewed changes
doc/frameworks/tensorflow/deploying_tensorflow_serving.rst Outdated
@@ -64,7 +64,7 @@ If you already have existing model artifacts in S3, you can skip training and de

from sagemaker.tensorflow import TensorFlowModel

model = TensorFlowModel(model_data='s3://mybucket/model.tar.gz', role='MySageMakerRole')
model = TensorFlowModel(model_data='s3://mybucket/model.tar.gz', role='MySageMakerRole', framework_version='MyFrameworkVersion')
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

is framework_version expect an int? can we do a dummy int 0.0.0 or x.x.x

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

updated to x.x.x

@pintaoz-aws pintaoz-aws requested a review from benieric February 14, 2025 20:01
@pintaoz-aws pintaoz-aws merged commit 5682c42 into aws:master Feb 17, 2025
12 of 14 checks passed
@pintaoz-aws pintaoz-aws mentioned this pull request Feb 17, 2025
Closed
evakravi pushed a commit to evakravi/sagemaker-python-sdk that referenced this pull request Mar 20, 2025
@pintaoz-aws @sagemaker-bot
Add framework_version to all TensorFlowModel examples (aws#5038)
c688624 
* Add framework_version to all TensorFlowModel examples

* update framework_version to x.x.x

---------

Co-authored-by: pintaoz <[email protected]>
pravali96 pushed a commit to pravali96/sagemaker-python-sdk that referenced this pull request Apr 21, 2025
@pintaoz-aws
Add framework_version to all TensorFlowModel examples (aws#5038)
b485f94 
* Add framework_version to all TensorFlowModel examples

* update framework_version to x.x.x

---------

Co-authored-by: pintaoz <[email protected]>
uyoldas pushed a commit to uyoldas/sagemaker-python-sdk that referenced this pull request May 23, 2025
@pintaoz-aws
Add framework_version to all TensorFlowModel examples (aws#5038)
0e18d2f 
* Add framework_version to all TensorFlowModel examples

* update framework_version to x.x.x

---------

Co-authored-by: pintaoz <[email protected]>
sage-maker added a commit that referenced this pull request Jul 1, 2025
@uyoldas @pintaoz-aws
@sage-maker @timkuo-amazon @zhaoqizqwang @sagemaker-bot
feature: integrate amtviz for visualization of tuning jobs (#5044)
5164316 
* feature: integrate amtviz for visualization of tuning jobs

* Move RecordSerializer and RecordDeserializer to sagemaker.serializers and sagemaker.deserialzers (#5037)

* Move RecordSerializer and RecordDeserializer to sagemaker.serializers and sagemaker.deserializers

* fix codestyle

* fix test

---------

Co-authored-by: pintaoz <[email protected]>

* Add framework_version to all TensorFlowModel examples (#5038)

* Add framework_version to all TensorFlowModel examples

* update framework_version to x.x.x

---------

Co-authored-by: pintaoz <[email protected]>

* Fix hyperparameter strategy docs (#5045)

* fix: pass in inference_ami_version to model_based endpoint type (#5043)

* fix: pass in inference_ami_version to model_based endpoint type

* documentation: update contributing.md w/ venv instructions and pip install fixes

---------

Co-authored-by: Zhaoqi <[email protected]>

* Add warning about not supporting torch.nn.SyncBatchNorm (#5046)

* Add warning about not supporting

* update wording

---------

Co-authored-by: pintaoz <[email protected]>

* prepare release v2.239.2

* update development version to v2.239.3.dev0

* change: update image_uri_configs  02-19-2025 06:18:15 PST

* fix: codestyle, type hints, license, and docstrings

* documentation: add docstring for amtviz module

* fix: fix docstyle and flake8 errors

* fix: code reformat using black

---------

Co-authored-by: Uemit Yoldas <[email protected]>
Co-authored-by: pintaoz-aws <[email protected]>
Co-authored-by: pintaoz <[email protected]>
Co-authored-by: parknate@ <[email protected]>
Co-authored-by: timkuo-amazon <[email protected]>
Co-authored-by: Zhaoqi <[email protected]>
Co-authored-by: ci <ci>
Co-authored-by: sagemaker-bot <[email protected]>
HollowTube pushed a commit to HollowTube/sagemaker-python-sdk that referenced this pull request Jul 15, 2025
@uyoldas @pintaoz-aws
@sage-maker @timkuo-amazon @zhaoqizqwang @sagemaker-bot
feature: integrate amtviz for visualization of tuning jobs (aws#5044)
abf7367 
* feature: integrate amtviz for visualization of tuning jobs

* Move RecordSerializer and RecordDeserializer to sagemaker.serializers and sagemaker.deserialzers (aws#5037)

* Move RecordSerializer and RecordDeserializer to sagemaker.serializers and sagemaker.deserializers

* fix codestyle

* fix test

---------

Co-authored-by: pintaoz <[email protected]>

* Add framework_version to all TensorFlowModel examples (aws#5038)

* Add framework_version to all TensorFlowModel examples

* update framework_version to x.x.x

---------

Co-authored-by: pintaoz <[email protected]>

* Fix hyperparameter strategy docs (aws#5045)

* fix: pass in inference_ami_version to model_based endpoint type (aws#5043)

* fix: pass in inference_ami_version to model_based endpoint type

* documentation: update contributing.md w/ venv instructions and pip install fixes

---------

Co-authored-by: Zhaoqi <[email protected]>

* Add warning about not supporting torch.nn.SyncBatchNorm (aws#5046)

* Add warning about not supporting

* update wording

---------

Co-authored-by: pintaoz <[email protected]>

* prepare release v2.239.2

* update development version to v2.239.3.dev0

* change: update image_uri_configs  02-19-2025 06:18:15 PST

* fix: codestyle, type hints, license, and docstrings

* documentation: add docstring for amtviz module

* fix: fix docstyle and flake8 errors

* fix: code reformat using black

---------

Co-authored-by: Uemit Yoldas <[email protected]>
Co-authored-by: pintaoz-aws <[email protected]>
Co-authored-by: pintaoz <[email protected]>
Co-authored-by: parknate@ <[email protected]>
Co-authored-by: timkuo-amazon <[email protected]>
Co-authored-by: Zhaoqi <[email protected]>
Co-authored-by: ci <ci>
Co-authored-by: sagemaker-bot <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@chad119 chad119 chad119 approved these changes

@Aditi2424 Aditi2424 Awaiting requested review from Aditi2424 Aditi2424 is a code owner automatically assigned from aws/sagemaker-ml-frameworks

@benieric benieric Awaiting requested review from benieric

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@pintaoz-aws @chad119 @benieric