@@ -57,28 +57,28 @@ jobs:
57
57
steps :
58
58
- uses : actions/checkout@v5
59
59
60
- - uses : actions/setup-python@v5
60
+ - name : Install uv and set Python version
61
+ uses : astral-sh/setup-uv@v6
61
62
with :
62
- # Python version here needs to be supported by all PL versions listed in back-compatible-versions.txt.
63
63
python-version : " 3.9"
64
64
65
65
- name : Install PL from source
66
66
env :
67
67
PACKAGE_NAME : pytorch
68
68
FREEZE_REQUIREMENTS : 1
69
69
timeout-minutes : 20
70
- run : pip install . --extra-index-url="${TORCH_URL}"
70
+ run : uv pip install . --extra-index-url="${TORCH_URL}"
71
71
if : inputs.pl_version == ''
72
72
73
73
- name : Install PL version
74
74
timeout-minutes : 20
75
- run : pip install "pytorch-lightning==${{ inputs.pl_version }}" --extra-index-url="${TORCH_URL}"
75
+ run : uv pip install "pytorch-lightning==${{ inputs.pl_version }}" --extra-index-url="${TORCH_URL}"
76
76
if : inputs.pl_version != ''
77
77
78
78
- name : Adjust tests -> PL
79
79
if : ${{ matrix.pkg-name != 'lightning' }}
80
80
run : |
81
- pip install -q -r .actions/requirements.txt
81
+ uv pip install -q -r .actions/requirements.txt
82
82
python .actions/assistant.py copy_replace_imports --source_dir="./tests" \
83
83
--source_import="lightning.fabric,lightning.pytorch" \
84
84
--target_import="lightning_fabric,pytorch_lightning"
@@ -115,7 +115,7 @@ jobs:
115
115
# export to env bool if secrets.AWS_REGION is not empty
116
116
run : echo "WITH_SECRETS=$([ -n '${{ secrets.AWS_REGION }}' ] && echo 1 || echo 0)" >> $GITHUB_ENV
117
117
118
- - run : pip install -r requirements/ci.txt
118
+ - run : uv pip install -r requirements/ci.txt
119
119
- name : Upload checkpoints to S3
120
120
if : ${{ env.WITH_SECRETS == '1' }}
121
121
working-directory : ${{ env.LEGACY_FOLDER }}
0 commit comments