bb

Author: Borda

.azure/gpu-tests-fabric.yml

@@ -114,17 +114,17 @@ jobs:
           pip install setuptools==75.6.0 jsonargparse==4.35.0
         displayName: "Install package & dependencies"
 
-      - bash: |
-          python -c "import bitsandbytes"
-        displayName: "Handle bitsandbytes"
-
       - bash: |
           set -e
           python requirements/collect_env_details.py
           python -c "import torch ; mgpu = torch.cuda.device_count() ; assert mgpu == 2, f'GPU: {mgpu}'"
           python requirements/pytorch/check-avail-extras.py
         displayName: "Env details"
 
+      - bash: |
+          python -c "import bitsandbytes"
+        displayName: "Handle bitsandbytes"
+
       - bash: python -m pytest lightning_fabric
         workingDirectory: src
         # without succeeded this could run even if the job has already failed

.azure/gpu-tests-pytorch.yml

@@ -128,17 +128,17 @@ jobs:
         condition: and(succeeded(), eq(variables['PACKAGE_NAME'], 'lightning'))
         displayName: "Drop PL for LAI"
 
-      - bash: |
-          python -c "import bitsandbytes"
-        displayName: "Handle bitsandbytes"
-
       - bash: |
           set -e
           python requirements/collect_env_details.py
           python -c "import torch ; mgpu = torch.cuda.device_count() ; assert mgpu == 2, f'GPU: {mgpu}'"
           python requirements/pytorch/check-avail-extras.py
         displayName: "Env details"
 
+      - bash: |
+          python -c "import bitsandbytes"
+        displayName: "Handle bitsandbytes"
+
       - bash: python -m pytest pytorch_lightning
         workingDirectory: src
         # without succeeded this could run even if the job has already failed