diff --git a/sdks/python/tox.ini b/sdks/python/tox.ini index 3b9d8531fb9e..e8402b87055f 100644 --- a/sdks/python/tox.ini +++ b/sdks/python/tox.ini @@ -390,9 +390,11 @@ commands = [testenv:py39-tensorflow-213] deps = - 212: tensorflow>=2.13rc1,<2.14 - # Help pip resolve conflict with typing-extensions for old version of TF https://github.com/apache/beam/issues/30852 - 212: pydantic<2.7 + 213: + tensorflow>=2.13rc1,<2.14 + # 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 commands = # Log tensorflow version for debugging @@ -419,12 +421,21 @@ commands = [testenv:py{39,310}-transformers-{428,429,430}] deps = - 428: transformers>=4.28.0,<4.29.0 - 429: transformers>=4.29.0,<4.30.0 - 430: transformers>=4.30.0,<4.31.0 - torch>=1.9.0,<1.14.0 - tensorflow==2.13.0 - protobuf==4.25.5 + 428: + transformers>=4.28.0,<4.29.0 + torch>=1.9.0,<1.14.0 + tensorflow==2.12.0 + protobuf==4.25.5 + 429: + transformers>=4.29.0,<4.30.0 + torch>=1.9.0,<1.14.0 + tensorflow==2.12.0 + protobuf==4.25.5 + 430: + transformers>=4.30.0,<4.31.0 + torch>=1.9.0,<1.14.0 + tensorflow==2.12.0 + protobuf==4.25.5 extras = test,gcp commands = # Log transformers and its dependencies version for debugging