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