Skip to content

Commit ac35062

Browse files
authored
Pin torch dependencies to tested versions (#35431)
1 parent 0595584 commit ac35062

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

sdks/python/setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -571,7 +571,7 @@ def get_portability_package_data():
571571
# in https://github.com/apache/beam/blob/master/sdks/python/tox.ini
572572
# For more info, see
573573
# https://docs.google.com/document/d/1c84Gc-cZRCfrU8f7kWGsNR2o8oSRjCM-dGHO9KvPWPw/edit?usp=sharing
574-
'torch': ['torch<=1.13.0,<2.1.0'],
574+
'torch': ['torch>=1.9.0,<2.1.0'],
575575
'tensorflow': ['tensorflow>=2.12rc1,<2.13'],
576576
'transformers': [
577577
'transformers>=4.28.0,<4.49.0',

0 commit comments

Comments
 (0)