Skip to content

Commit

Permalink
Pin protobuf version for TF dependency tests (apache#32719)
Browse files Browse the repository at this point in the history
* Pin protobuf version for TF dependency tests

* pin specfic protobuf version

* increment tested TF version

* pin at higher version of protobuf

* fix incorrect configuration

* try bumping tf version

* further specify dependency versions

* try adding other ml testing deps for compat

* fix transformers tests

* tweak deps for transformers

* whitespace
  • Loading branch information
jrmccluskey authored and reeba212 committed Dec 4, 2024
1 parent cfa5c63 commit 772d5af
Showing 1 changed file with 8 additions and 5 deletions.
13 changes: 8 additions & 5 deletions sdks/python/tox.ini
Original file line number Diff line number Diff line change
Expand Up @@ -394,10 +394,12 @@ commands =

[testenv:py39-tensorflow-212]
deps =
212: tensorflow>=2.12rc1,<2.13
# Help pip resolve conflict with typing-extensions for old version of TF https://github.com/apache/beam/issues/30852
212: pydantic<2.7
extras = test,gcp
212:
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,ml_test
commands =
# Log tensorflow version for debugging
/bin/sh -c "pip freeze | grep -E tensorflow"
Expand Down Expand Up @@ -428,7 +430,8 @@ deps =
430: transformers>=4.30.0,<4.31.0
torch>=1.9.0,<1.14.0
tensorflow==2.12.0
extras = test,gcp
protobuf==4.25.5
extras = test,gcp,ml_test
commands =
# Log transformers and its dependencies version for debugging
/bin/sh -c "pip freeze | grep -E transformers"
Expand Down

0 comments on commit 772d5af

Please sign in to comment.