(beam) branch users/damccorm/torch-cpu-only created (now 21cd6a886cf)

[da...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

damccorm pushed a change to branch users/damccorm/torch-cpu-only
in repository https://gitbox.apache.org/repos/asf/beam.git


      at 21cd6a886cf [WIP] Force torch from cpu wheels

This branch includes the following new commits:

     new 21cd6a886cf [WIP] Force torch from cpu wheels

The 1 revisions listed above as "new" are entirely new to this
repository and will be described in separate emails.  The revisions
listed as "add" were already present in the repository and have only
been added to this reference.

(beam) 01/01: [WIP] Force torch from cpu wheels

[da...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

damccorm pushed a commit to branch users/damccorm/torch-cpu-only
in repository https://gitbox.apache.org/repos/asf/beam.git

commit 21cd6a886cfb8469497ef36bc828699556436a65
Author: Danny Mccormick <da...@google.com>
AuthorDate: Tue Oct 21 13:41:53 2025 -0400

    [WIP] Force torch from cpu wheels
---
 sdks/python/setup.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/sdks/python/setup.py b/sdks/python/setup.py
index 719d188ed26..8eb69876dd1 100644
--- a/sdks/python/setup.py
+++ b/sdks/python/setup.py
@@ -422,6 +422,8 @@ if __name__ == '__main__':
           # Dependabot won't be able to parse the main list. Any dynamic
           # dependencies will not receive updates from Dependabot.
       ] + pyarrow_dependency,
+      # Force torch dependencies to be pulled from the PyTorch CPU wheel repository
+      find_links=['https://download.pytorch.org/whl/cpu'],
       python_requires=python_requires,
       # BEAM-8840: Do NOT use tests_require or setup_requires.
       extras_require={