From 8c3f9c22508b1afac3d23f5e2a754a152013e404 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Wed, 21 Jan 2026 14:01:10 +0000 Subject: [PATCH] Remove pinning against torch version --- setup.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index 06b938bb31..8030c5408e 100644 --- a/setup.py +++ b/setup.py @@ -86,8 +86,12 @@ def _main(): print("-- Git branch:", branch) print("-- Git SHA:", sha) print("-- Git tag:", tag) - pytorch_package_dep = _get_pytorch_version() - print("-- PyTorch dependency:", pytorch_package_dep) + # This used to be passed to install_requires + # which would cause pinning against a specific torch version in releases. + # I don't think we want to pin at all? + # TODO: revisit if needed. Maybe it's needed for nightlies. Unsure. + # pytorch_package_dep = _get_pytorch_version() + # print("-- PyTorch dependency:", pytorch_package_dep) version = _get_version(sha) print("-- Building version", version) @@ -135,7 +139,7 @@ def _main(): "build_ext": setup_helpers.get_build_ext(), "clean": clean, }, - install_requires=[pytorch_package_dep], + install_requires=[], zip_safe=False, )