From 56b0b22abd35c9d1d9e2d3fc87e4907eea622f8a Mon Sep 17 00:00:00 2001
From: Danny McCormick <dannymccormick@google.com>
Date: Wed, 25 Jun 2025 11:12:24 -0400
Subject: [PATCH] Pin torch dependencies to tested versions

---
 sdks/python/setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sdks/python/setup.py b/sdks/python/setup.py
index 6c799ba943bb..a0bbc301435b 100644
--- a/sdks/python/setup.py
+++ b/sdks/python/setup.py
@@ -571,7 +571,7 @@ def get_portability_package_data():
           # in https://github.com/apache/beam/blob/master/sdks/python/tox.ini
           # For more info, see
           # https://docs.google.com/document/d/1c84Gc-cZRCfrU8f7kWGsNR2o8oSRjCM-dGHO9KvPWPw/edit?usp=sharing
-          'torch': ['torch<=1.13.0,<2.1.0'],
+          'torch': ['torch>=1.9.0,<2.1.0'],
           'tensorflow': ['tensorflow>=2.12rc1,<2.13'],
           'transformers': [
               'transformers>=4.28.0,<4.49.0',