x

Lightning-AI · Apr 26, 2024 · b501743 · b501743
1 parent a18fa66
commit b501743
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/.azure/gpu-tests-fabric.yml b/.azure/gpu-tests-fabric.yml
@@ -101,9 +101,7 @@ jobs:
           done
         displayName: "Adjust dependencies"
 
-        # TODO: remove
       - bash: |
-          pip install torch==2.3.0 torchvision --index-url https://download.pytorch.org/whl/test/cu121
           extra=$(python -c "print({'lightning': 'fabric-'}.get('$(PACKAGE_NAME)', ''))")
           pip install -e ".[${extra}dev]" pytest-timeout -U --find-links="${TORCH_URL}"
         displayName: "Install package & dependencies"

diff --git a/.azure/gpu-tests-pytorch.yml b/.azure/gpu-tests-pytorch.yml
@@ -104,9 +104,7 @@ jobs:
           done
         displayName: "Adjust dependencies"
 
-      # TODO: remove
       - bash: |
-          pip install torch==2.3.0 torchvision --index-url https://download.pytorch.org/whl/test/cu121
           extra=$(python -c "print({'lightning': 'pytorch-'}.get('$(PACKAGE_NAME)', ''))")
           pip install -e ".[${extra}dev]" pytest-timeout -U --find-links="${TORCH_URL}"
         displayName: "Install package & dependencies"