future

Author: Borda

.azure/gpu-tests-fabric.yml

@@ -117,10 +117,9 @@ jobs:
         displayName: "Install package & dependencies"
 
       - bash: |
-          # get pytorch version
-          PYTORCH_VERSION=$(python -c "import torch; print(torch.__version__.split('+')[0])")
           # FixMe: uninstall bitsandbytes for pytorch 2.6 as it is not compatible with `triton.ops`
-          if [[ "${PYTORCH_VERSION}" == "2.6.0" ]]; then
+          # if the job name include key word future, then uninstall bitsandbytes
+          if [[ "$(Agent.JobName)" == *"future"* ]]; then
             pip uninstall -y bitsandbytes
           else
             python -c "import bitsandbytes"

.azure/gpu-tests-pytorch.yml

@@ -131,10 +131,9 @@ jobs:
         displayName: "Drop PL for LAI"
 
       - bash: |
-          # get pytorch version
-          PYTORCH_VERSION=$(python -c "import torch; print(torch.__version__.split('+')[0])")
           # FixMe: uninstall bitsandbytes for pytorch 2.6 as it is not compatible with `triton.ops`
-          if [[ "${PYTORCH_VERSION}" == "2.6.0" ]]; then
+          # if the job name include key word future, then uninstall bitsandbytes
+          if [[ "$(Agent.JobName)" == *"future"* ]]; then
             pip uninstall -y bitsandbytes
           else
             python -c "import bitsandbytes"