From 030f36bcfa99298ac725d50a150eb197aa988467 Mon Sep 17 00:00:00 2001
From: Luca Antiga <luca.antiga@gmail.com>
Date: Tue, 10 Dec 2024 23:31:31 +0100
Subject: [PATCH] Pin setuptools in GPU tests (#20489)

---
 .azure/gpu-benchmarks.yml            | 4 +++-
 .azure/gpu-tests-fabric.yml          | 1 +
 .azure/gpu-tests-pytorch.yml         | 1 +
 docs/source-pytorch/common/index.rst | 3 ++-
 4 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/.azure/gpu-benchmarks.yml b/.azure/gpu-benchmarks.yml
index 24b78542a798a..6825850ae01bd 100644
--- a/.azure/gpu-benchmarks.yml
+++ b/.azure/gpu-benchmarks.yml
@@ -75,7 +75,9 @@ jobs:
           pip list
         displayName: "Image info & NVIDIA"
 
-      - bash: pip install -e .[dev] --find-links ${TORCH_URL}
+      - bash: |
+          pip install -e .[dev] --find-links ${TORCH_URL}
+          pip install setuptools==75.6.0
         env:
           FREEZE_REQUIREMENTS: "1"
         displayName: "Install package"
diff --git a/.azure/gpu-tests-fabric.yml b/.azure/gpu-tests-fabric.yml
index cd1cf09373c6c..9d7514f285bda 100644
--- a/.azure/gpu-tests-fabric.yml
+++ b/.azure/gpu-tests-fabric.yml
@@ -107,6 +107,7 @@ jobs:
       - bash: |
           extra=$(python -c "print({'lightning': 'fabric-'}.get('$(PACKAGE_NAME)', ''))")
           pip install -e ".[${extra}dev]" pytest-timeout -U --find-links="${TORCH_URL}" --find-links="${TORCHVISION_URL}"
+          pip install setuptools==75.6.0
         displayName: "Install package & dependencies"
 
       - bash: |
diff --git a/.azure/gpu-tests-pytorch.yml b/.azure/gpu-tests-pytorch.yml
index ae4c1367b646e..b48961c7f3a26 100644
--- a/.azure/gpu-tests-pytorch.yml
+++ b/.azure/gpu-tests-pytorch.yml
@@ -111,6 +111,7 @@ jobs:
       - bash: |
           extra=$(python -c "print({'lightning': 'pytorch-'}.get('$(PACKAGE_NAME)', ''))")
           pip install -e ".[${extra}dev]" pytest-timeout -U --find-links="${TORCH_URL}" --find-links="${TORCHVISION_URL}"
+          pip install setuptools==75.6.0
         displayName: "Install package & dependencies"
 
       - bash: pip uninstall -y lightning
diff --git a/docs/source-pytorch/common/index.rst b/docs/source-pytorch/common/index.rst
index 42f7adcc2ed24..e0492a7e747fe 100644
--- a/docs/source-pytorch/common/index.rst
+++ b/docs/source-pytorch/common/index.rst
@@ -23,6 +23,7 @@
    ../data/data
    ../model/own_your_loop
    ../advanced/model_init
+   ../common/tbptt
 
 
 #############
@@ -205,7 +206,7 @@ How-to Guides
 .. displayitem::
     :header: Truncated Back-Propagation Through Time
     :description: Efficiently step through time when training recurrent models
-    :button_link: ../common/tbtt.html
+    :button_link: ../common/tbptt.html
     :col_css: col-md-4
     :height: 180