@@ -57,28 +57,28 @@ jobs:
5757 steps :
5858 - uses : actions/checkout@v5
5959
60- - uses : actions/setup-python@v5
60+ - name : Install uv and set Python version
61+ uses : astral-sh/setup-uv@v6
6162 with :
62- # Python version here needs to be supported by all PL versions listed in back-compatible-versions.txt.
6363 python-version : " 3.9"
6464
6565 - name : Install PL from source
6666 env :
6767 PACKAGE_NAME : pytorch
6868 FREEZE_REQUIREMENTS : 1
6969 timeout-minutes : 20
70- run : pip install . --extra-index-url="${TORCH_URL}"
70+ run : uv pip install . --extra-index-url="${TORCH_URL}"
7171 if : inputs.pl_version == ''
7272
7373 - name : Install PL version
7474 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}"
7676 if : inputs.pl_version != ''
7777
7878 - name : Adjust tests -> PL
7979 if : ${{ matrix.pkg-name != 'lightning' }}
8080 run : |
81- pip install -q -r .actions/requirements.txt
81+ uv pip install -q -r .actions/requirements.txt
8282 python .actions/assistant.py copy_replace_imports --source_dir="./tests" \
8383 --source_import="lightning.fabric,lightning.pytorch" \
8484 --target_import="lightning_fabric,pytorch_lightning"
@@ -115,7 +115,7 @@ jobs:
115115 # export to env bool if secrets.AWS_REGION is not empty
116116 run : echo "WITH_SECRETS=$([ -n '${{ secrets.AWS_REGION }}' ] && echo 1 || echo 0)" >> $GITHUB_ENV
117117
118- - run : pip install -r requirements/ci.txt
118+ - run : uv pip install -r requirements/ci.txt
119119 - name : Upload checkpoints to S3
120120 if : ${{ env.WITH_SECRETS == '1' }}
121121 working-directory : ${{ env.LEGACY_FOLDER }}
0 commit comments