Skip to content

Conversation

@NihalHarish
Copy link
Contributor

Description of changes:

  • Simple caching logic to avoid the repeated usage of version.parse(tf.__version__)

Style and formatting:

I have run pre-commit install to ensure that auto-formatting happens with every commit.

Issue number, if available

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

Copy link
Contributor

@connorgoggins connorgoggins left a comment

Choose a reason for hiding this comment

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

Non-blocking: might also be worth updating smdebug/tensorflow/version_check.py, tests/tensorflow2/utils.py, tests/zero_code_change/smdataparallel_tests/tensorflow2/test_tf2_smdataparallel.py, tests/zero_code_change/test_tensorflow2_integration.py, etc. to use the cached version as well.

Copy link
Contributor

@ndodda-amazon ndodda-amazon left a comment

Choose a reason for hiding this comment

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

Changes look good! Can we cache the PyTorch versions as well for consistency?

@NihalHarish NihalHarish merged commit 5bf9530 into master Jan 13, 2021
@NihalHarish NihalHarish deleted the cache_tf_versions branch January 13, 2021 23:28
sophiayue1116 pushed a commit to sophiayue1116/sagemaker-debugger that referenced this pull request Jan 20, 2021
sophiayue1116 pushed a commit to sophiayue1116/sagemaker-debugger that referenced this pull request Jan 20, 2021
sophiayue1116 pushed a commit to sophiayue1116/sagemaker-debugger that referenced this pull request Jan 20, 2021
sophiayue1116 pushed a commit to sophiayue1116/sagemaker-debugger that referenced this pull request Jan 20, 2021
sophiayue1116 pushed a commit to sophiayue1116/sagemaker-debugger that referenced this pull request Jan 22, 2021
NihalHarish added a commit that referenced this pull request Jan 25, 2021
sophiayue1116 pushed a commit to sophiayue1116/sagemaker-debugger that referenced this pull request Jan 29, 2021
sophiayue1116 pushed a commit to sophiayue1116/sagemaker-debugger that referenced this pull request Jan 29, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants