Skip to content

Commit 030f36b

Browse files
authored
Pin setuptools in GPU tests (#20489)
1 parent 5495204 commit 030f36b

File tree

4 files changed

+7
-2
lines changed

4 files changed

+7
-2
lines changed

.azure/gpu-benchmarks.yml

+3-1
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,9 @@ jobs:
7575
pip list
7676
displayName: "Image info & NVIDIA"
7777
78-
- bash: pip install -e .[dev] --find-links ${TORCH_URL}
78+
- bash: |
79+
pip install -e .[dev] --find-links ${TORCH_URL}
80+
pip install setuptools==75.6.0
7981
env:
8082
FREEZE_REQUIREMENTS: "1"
8183
displayName: "Install package"

.azure/gpu-tests-fabric.yml

+1
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,7 @@ jobs:
107107
- bash: |
108108
extra=$(python -c "print({'lightning': 'fabric-'}.get('$(PACKAGE_NAME)', ''))")
109109
pip install -e ".[${extra}dev]" pytest-timeout -U --find-links="${TORCH_URL}" --find-links="${TORCHVISION_URL}"
110+
pip install setuptools==75.6.0
110111
displayName: "Install package & dependencies"
111112
112113
- bash: |

.azure/gpu-tests-pytorch.yml

+1
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,7 @@ jobs:
111111
- bash: |
112112
extra=$(python -c "print({'lightning': 'pytorch-'}.get('$(PACKAGE_NAME)', ''))")
113113
pip install -e ".[${extra}dev]" pytest-timeout -U --find-links="${TORCH_URL}" --find-links="${TORCHVISION_URL}"
114+
pip install setuptools==75.6.0
114115
displayName: "Install package & dependencies"
115116
116117
- bash: pip uninstall -y lightning

docs/source-pytorch/common/index.rst

+2-1
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
../data/data
2424
../model/own_your_loop
2525
../advanced/model_init
26+
../common/tbptt
2627

2728

2829
#############
@@ -205,7 +206,7 @@ How-to Guides
205206
.. displayitem::
206207
:header: Truncated Back-Propagation Through Time
207208
:description: Efficiently step through time when training recurrent models
208-
:button_link: ../common/tbtt.html
209+
:button_link: ../common/tbptt.html
209210
:col_css: col-md-4
210211
:height: 180
211212

0 commit comments

Comments
 (0)