tweak deps for transformers

apache · Oct 11, 2024 · 99b6906 · 99b6906
1 parent 9fcd638
commit 99b6906
Showing 1 changed file with 4 additions and 10 deletions.
diff --git a/sdks/python/tox.ini b/sdks/python/tox.ini
@@ -390,7 +390,7 @@ commands =
 
 [testenv:py39-tensorflow-212]
 deps =
- 213: 
+ 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
@@ -423,19 +423,13 @@ commands =
 deps =
  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
+ torch>=1.9.0,<1.14.0
+ tensorflow==2.12.0
+ protobuf==4.25.5
 extras = test,gcp,ml_test
 commands =
  # Log transformers and its dependencies version for debugging