try adding other ml testing deps for compat

apache · Oct 11, 2024 · 8cfca1f · 8cfca1f
1 parent 6c8a536
commit 8cfca1f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/sdks/python/test-suites/tox/py39/build.gradle b/sdks/python/test-suites/tox/py39/build.gradle
@@ -149,9 +149,9 @@ postCommitPyDep.dependsOn "testPy39tft-113"
 // postCommitPyDep.dependsOn "testPy39onnx-113"
 
 // Create a test task for each minor version of tensorflow
-toxTask "testPy39tensorflow-213", "py39-tensorflow-213", "${posargs}"
-test.dependsOn "testPy39tensorflow-213"
-postCommitPyDep.dependsOn "testPy39tensorflow-213"
+toxTask "testPy39tensorflow-212", "py39-tensorflow-212", "${posargs}"
+test.dependsOn "testPy39tensorflow-212"
+postCommitPyDep.dependsOn "testPy39tensorflow-212"
 
 // Create a test task for each minor version of transformers
 toxTask "testPy39transformers-428", "py39-transformers-428", "${posargs}"

diff --git a/sdks/python/tox.ini b/sdks/python/tox.ini
@@ -388,14 +388,14 @@ commands =
   # Run all ONNX unit tests
   pytest -o junit_suite_name={envname} --junitxml=pytest_{envname}.xml -n 6 -m uses_onnx {posargs}
 
-[testenv:py39-tensorflow-213]
+[testenv:py39-tensorflow-212]
 deps =
   213: 
-    tensorflow>=2.13rc1,<2.14
+    tensorflow>=2.12rc1,<2.13
     # Help pip resolve conflict with typing-extensions for old version of TF https://github.com/apache/beam/issues/30852
     pydantic<2.7
     protobuf==4.25.5
-extras = test,gcp
+extras = test,gcp,ml_test
 commands =
   # Log tensorflow version for debugging
   /bin/sh -c "pip freeze | grep -E tensorflow"