File tree Expand file tree Collapse file tree 2 files changed +9
-2
lines changed Expand file tree Collapse file tree 2 files changed +9
-2
lines changed Original file line number Diff line number Diff line change 1212import subprocess
1313import sys
1414
15+ from torch_pin import NIGHTLY_VERSION , TORCH_VERSION
16+
1517
1618def python_is_compatible ():
1719 # Scrape the version range from pyproject.toml, which should be in the current directory.
@@ -71,7 +73,6 @@ def python_is_compatible():
7173#
7274# NOTE: If you're changing, make the corresponding change in .ci/docker/ci_commit_pins/pytorch.txt
7375# by picking the hash from the same date in https://hud.pytorch.org/hud/pytorch/pytorch/nightly/
74- NIGHTLY_VERSION = "dev20250906"
7576
7677
7778def install_requirements (use_pytorch_nightly ):
@@ -89,7 +90,11 @@ def install_requirements(use_pytorch_nightly):
8990 # Setting use_pytorch_nightly to false to test the pinned PyTorch commit. Note
9091 # that we don't need to set any version number there because they have already
9192 # been installed on CI before this step, so pip won't reinstall them
92- "torch==2.9.0" if use_pytorch_nightly else "torch" ,
93+ (
94+ f"torch=={ TORCH_VERSION } .{ NIGHTLY_VERSION } "
95+ if use_pytorch_nightly
96+ else "torch"
97+ ),
9398 ]
9499
95100 # Install the requirements for core ExecuTorch package.
Original file line number Diff line number Diff line change 1+ TORCH_VERSION = "2.9.0"
2+ NIGHTLY_VERSION = "dev20250906"
You can’t perform that action at this time.
0 commit comments