-
Notifications
You must be signed in to change notification settings - Fork 6
Upgrade to TensorFlow 2.18.0+ and Python 3.12, add TensorFlow support to pipeline models #561
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.
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
base: main
Are you sure you want to change the base?
Conversation
As this is a breaking change, we may need to bump up the aiverify-test-engine as major release, e.g. |
Done |
Description
Upgrade to TensorFlow 2.18.0+ and Python 3.12, add TensorFlow support to pipeline models. Fixes: #326
Motivation and Context
This PR adds TensorFlow support to pipeline models and upgrades the TensorFlow version to 2.18.0+, which relaxes the Python version to 3.12. As discussed in #326 this is a breaking change because TensorFlow >=2.18 requires Keras 3 while TensorFlow < 2.18 requires Keras 2 with a different set of Keras APIs.
If the user wishes to run older versions of TensorFlow, they should run it as a pipeline and implement the
predict
function to call the appropriate Keras API, e.g. line 31 below:aiverify/stock-plugins/user_defined_files/pipeline/sample_fashion_mnist_tensorflow/fashionCustomClass.py
Lines 22 to 34 in 906738c
Type of Change
How to Test
Tested:
aiverify-apigw
)aiverify-portal
Note:
aiverify-apigw
requiresPython>=3.11
therefore is not tested in Python 3.10aiverify-test-engine/tests/io/test_tensorflowpipeline.py
is failing because the required asset is not available until this PR is merged intomain
branchChecklist
Please check all the boxes that apply to this pull request using "x":
Screenshots (if applicable)
[If the changes involve visual modifications, include screenshots or GIFs that demonstrate the changes.]
Additional Notes
[Add any additional information or context that might be relevant to reviewers.]
Developer Certificate of Origin