Skip to content

Commit 47167b6

Browse files
committed
update
1 parent fba4586 commit 47167b6

File tree

2 files changed

+10
-8
lines changed

2 files changed

+10
-8
lines changed

.azure/gpu-tests-fabric.yml

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -99,13 +99,14 @@ jobs:
9999
displayName: "Image info & NVIDIA"
100100
101101
- bash: |
102-
cd requirements/fabric
103-
pip install -U "lightning-utilities[cli]"
104-
python -m lightning_utilities.cli requirements set-oldest --req_files "['base.txt', 'strategies.txt']"
102+
set -ex
105103
pip install "cython<3.0" wheel # for compatibility
106104
pip install -U "lightning-utilities[cli]"
105+
cd requirements/fabric
106+
# replace range by pin minimal requirements
107+
python -m lightning_utilities.cli requirements set-oldest --req_files "['base.txt', 'strategies.txt']"
107108
# drop deepspeed since it is not supported by our minimal Torch requirements
108-
python -m lightning_utilities.cli requirements prune-pkgs --packages deepspeed --req_files requirements/fabric/strategies.txt
109+
python -m lightning_utilities.cli requirements prune-pkgs --packages deepspeed --req_files strategies.txt
109110
# uninstall deepspeed since some older docker images have it pre-installed
110111
pip uninstall -y deepspeed
111112
condition: contains(variables['Agent.JobName'], 'oldest')

.azure/gpu-tests-pytorch.yml

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -103,13 +103,14 @@ jobs:
103103
displayName: "Image info & NVIDIA"
104104
105105
- bash: |
106-
cd requirements/pytorch
107-
pip install -U "lightning-utilities[cli]"
108-
python -m lightning_utilities.cli requirements set-oldest --req_files "['base.txt', 'extra.txt', 'strategies.txt', 'examples.txt']"
106+
set -ex
109107
pip install "cython<3.0" wheel # for compatibility
110108
pip install -U "lightning-utilities[cli]"
109+
cd requirements/pytorch
110+
# replace range by pin minimal requirements
111+
python -m lightning_utilities.cli requirements set-oldest --req_files "['base.txt', 'extra.txt', 'strategies.txt', 'examples.txt']"
111112
# drop deepspeed since it is not supported by our minimal Torch requirements
112-
python -m lightning_utilities.cli requirements prune-pkgs --packages deepspeed --req_files requirements/pytorch/strategies.txt
113+
python -m lightning_utilities.cli requirements prune-pkgs --packages deepspeed --req_files strategies.txt
113114
# uninstall deepspeed since some older docker images have it pre-installed
114115
pip uninstall -y deepspeed
115116
condition: contains(variables['Agent.JobName'], 'oldest')

0 commit comments

Comments
 (0)